torch.complex

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

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

Параметры

• real (Тензор) – Вещественная часть комплексного тензора. Должна быть полуцелочисленной, плавающей или двойной точности.
• 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