PyTorch
torch / torch
torch.complex¶
-
torch.
complex
(real, imag, *, out=None) → Tensor¶ Constructs a complex tensor with its real part equal to
real
and its imaginary part equal toimag
.- Parameters
- Keyword Arguments
out (Tensor) – If the inputs are
torch.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) >>> z tensor([(1.+3.j), (2.+4.j)]) >>> z.dtype torch.complex64
此页内容是否对您有帮助
感谢反馈!