PyTorch生成复数tensor：torch.complex函数

1、torch.complex函数介绍

在PyTorch中我们可以生成复数tensor，方法是：

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

参数介绍：

real：生成的复数中的实部。
imag：生成的复数中的虚部。
out：输出的tensor。

提示：real实部和imag虚部必须类型，而且只能是float32或者double，也就是都是64位的。如果输入为torch.float32，则输出为torch.complet64。如果输入为torch.foat64，则输出为torch.complete128。

2、torch.complex函数代码

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