快捷方式

torch.view_as_complex

torch.view_as_complex(input) Tensor

input 作为复数张量返回一个视图。对于大小为 m1,m2,,mi,2m1, m2, \dots, mi, 2 的输入复数张量,此函数返回一个大小为 m1,m2,,mim1, m2, \dots, mi 的新复数张量,其中输入张量的最后一个维度预计表示复数的实部和虚部。

警告

view_as_complex() 仅支持 torch.dtypetorch.float64torch.float32 的张量。输入的最后一个维度的大小应为 2。此外,张量的最后一个维度的 步长 必须为 1。所有其他维度的步长必须为偶数。

参数

input (Tensor) – 输入张量。

示例

>>> 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)])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源