Python PyTorch complex用法及代碼示例

本文簡要介紹python語言中 torch.complex 的用法。

用法:

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

參數：

• real(Tensor) -複張量的實部。必須是浮點數或雙精度數。

• imag(Tensor) -複張量的虛部。必須與 real 具有相同的 dtype。

關鍵字參數：

out(Tensor) -如果輸入是 torch.float32 ，則必須是 torch.complex64 。如果輸入是 torch.float64 ，則必須是 torch.complex128 。

構造一個複數張量，其實部等於 real ，虛部等於 imag 。

例子：

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