torch.view_as_complex¶
- torch.view_as_complex(input) Tensor ¶
将
input
作为复数张量返回一个视图。对于大小为 的输入复数张量,此函数返回一个大小为 的新复数张量,其中输入张量的最后一个维度预计表示复数的实部和虚部。警告
view_as_complex()
仅支持torch.dtype
为torch.float64
和torch.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)])