用法:
torch.view_as_complex(input) → Tensor
参数:
input(Tensor) -输入张量。
将input
的视图作为复数张量返回。对于 size
m1,m2,…,mi,2 的输入复数张量,此函数返回 size
m1,m2,…,mi 的新复数张量,其中输入张量的最后一个维度应表示复数的实部和虚部。
警告
view_as_complex()仅支持带有 torch.dtype(torch.float64
和 torch.float32
)的张量。输入的最后一维应为size
2。此外,张量的最后一维的stride
必须为 1。所有其他维度的步幅必须是偶数。
例子:
>>> x=torch.randn(4, 2)
>>> x
tensor([[ 1.6116, -0.5772],
[-1.4606, -0.9120],
[ 0.0786, -1.7497],
[-0.6561, -1.6623]])
>>> torch.view_as_complex(x)
tensor([(1.6116-0.5772j), (-1.4606-0.9120j), (0.0786-1.7497j), (-0.6561-1.6623j)])