torch.complex

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

Создаёт комплексный тензор, действительная часть которого равна real, а мнимая часть равна imag.

Параметры:

• real (Tensor) – Действительная часть комплексного тензора. Тип должен быть half, float или double.
• imag (Tensor) – Мнимая часть комплексного тензора. Тип данных должен совпадать с типом данных real.

Именованные аргументы:

out (Tensor) – Если входные данные — torch.float32, значение должно быть torch.complex64. Если входные данные — torch.float64, значение должно быть torch.complex128.

Пример:

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