torch.complex#

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

Constructs a complex tensor with its real part equal toreal and itsimaginary part equal toimag.

Parameters

• real (Tensor) – The real part of the complex tensor. Must be half, float or double.

• imag (Tensor) – The imaginary part of the complex tensor. Must be same dtypeasreal.

Keyword Arguments

out (Tensor) – If the inputs aretorch.float32, must betorch.complex64. If the inputs aretorch.float64, must betorch.complex128.

Example:

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