Функция torch.polar в PyTorch
Справочное руководство PyTorch torch
torch.polar— это функция в PyTorch, используемая для создания комплексных тензоров из полярных координат.
Определение функции
torch.polar(abs, angle)
Пример использования
Пример
import torch
# Создать тензор абсолютных значений (модулей)
abs_val = torch.tensor([1.0, 2.0, 3.0])
# Создать тензор углов (в радианах)
angle = torch.tensor([0.0, torch.pi/2, torch.pi])
# Создать комплексный тензор из полярных координат
x = torch.polar(abs_val, angle)
print(x)
print("Абсолютное значение:", x.abs())
print("Угол:", x.angle())
# Создать тензор абсолютных значений (модулей)
abs_val = torch.tensor([1.0, 2.0, 3.0])
# Создать тензор углов (в радианах)
angle = torch.tensor([0.0, torch.pi/2, torch.pi])
# Создать комплексный тензор из полярных координат
x = torch.polar(abs_val, angle)
print(x)
print("Абсолютное значение:", x.abs())
print("Угол:", x.angle())
Другие расширения