torch.complex

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

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

Параметры:

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

Ключевые аргументы:

out (Тензор) – Если входные данные 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