pytorch每日一学26(torch.complex())生成复数tensor

第26个方法

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

• 此方法其实额前面讲过的一些方法有很大的关系，此方法为创建复数tensor的方法，其中real为实部，imag为虚部，real和imag必须位数相同，如果real和imag同为float32那么生成的complex就为complex64,。
  参数介绍：
• real：生成的复数中的实部。
• imag：生成的复数中的虚部。
• out：输出的tensor。

使用方法如下：

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