Функция PyTorch torch.angle
Справочное руководство Pytorch torch
torch.angleЭто функция в PyTorch, используемая для возврата фазового угла (аргумента) комплексного тензора. Возвращаемое значение выражено в радианах и находится в диапазоне [-pi, pi].
Определение функции
torch.angle(input, out=None)
Пример использования
Пример
import torch
# Создать комплексный тензор
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# Вернуть фазовый угол
result = torch.angle(x)
print(result)
# Создать комплексный тензор
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# Вернуть фазовый угол
result = torch.angle(x)
print(result)
Результат вывода:
tensor([ 0.7854, 0.0000, 1.5708, 3.1416])
Другие расширения