torch.view_as_complex

torch.view_as_complex(input) → 张量

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

警告

view_as_complex() 仅支持具有 torch.dtype torch.float64 和 torch.float32 的张量。输入的 最后一个维度的大小应为 size 2。此外， 张量的最后一个维度的 stride 必须为 1。所有其他维度的 步幅必须是偶数。

Parameters

输入 (张量) – 输入张量。

示例：

>>> x=torch.randn(4, 2)
>>> x
张量([[ 1.6116, -0.5772],
        [-1.4606, -0.9120],
        [ 0.0786, -1.7497],
        [-0.6561, -1.6623]])
>>> torch.view_as_complex(x)
张量([(1.6116-0.5772j), (-1.4606-0.9120j), (0.0786-1.7497j), (-0.6561-1.6623j)])