torch.complex

torch.complex(real, imag, *, out=None) → Tensor

构建一个复数张量，其实部为 real，虚部为 imag。

参数

• real (Tensor) – 表示复数张量的实部，且该实部必须是半精度、浮点或双精度类型。

• imag (Tensor) – 复数张量的虚部部分，必须与 real 同类型。

关键字参数

out (Tensor) – 如果输入是 torch.float32，则输出必须为 torch.complex64；如果输入是 torch.float64，则输出必须为 torch.complex128。

示例：

>>> real = torch.tensor([1, 2], dtype=torch.float32)
>>> imag = torch.tensor([3, 4], dtype=torch.float32)
>>> z = torch.complex(real, imag)
>>> z
tensor([(1.+3.j), (2.+4.j)])
>>> z.dtype
torch.complex64