Функция torch.imag в PyTorch
Справочное руководство Pytorch torch
torch.imag— это функция в PyTorch, используемая для получения мнимой части комплексного тензора.
Определение функции
torch.imag(input)
Пример использования
Пример
import torch
# Создать комплексный тензор
x = torch.complex(torch.tensor([1.0, 2.0]), torch.tensor([3.0, 4.0]))
# Получить мнимую часть
y = torch.imag(x)
print("Мнимая часть:", y)
# Создать комплексный тензор
x = torch.complex(torch.tensor([1.0, 2.0]), torch.tensor([3.0, 4.0]))
# Получить мнимую часть
y = torch.imag(x)
print("Мнимая часть:", y)
Результат вывода:
虚部: tensor([3., 4.])
Другие расширения