Функция PyTorch torch.complex


Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.complexявляется функцией в PyTorch, используемой для создания комплексных тензоров из вещественной и мнимой частей.

Определение функции

torch.complex(real, imag)

Пример использования

Пример

import torch

# Создать тензор вещественной части
real = torch.tensor([1.0, 2.0, 3.0])

# Создать тензор мнимой части
imag = torch.tensor([4.0, 5.0, 6.0])

# Создать комплексный тензор из действительной и мнимой частей
x = torch.complex(real, imag)

print(x)
print("Действительная часть:", x.real)
print("Мнимая часть:", x.imag)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения