Функция angle
Функция angle вычисляет фазовый угол (аргумент) для каждого элемента комплексного тензора.
Результат возвращается в радианах в диапазоне от -π до π.
На вход функция принимает комплексный тензор, созданный с помощью torch.complex
или преобразованный из действительного тензора.
Синтаксис
torch.angle(input, *, out=None)
Параметры:
-
input- комплексный тензор, для элементов которого вычисляется фаза. -
out(необязательный) - тензор для записи результата.
Пример
Давайте создадим комплексный тензор и вычислим фазу для его элементов:
import torch
t = torch.tensor([1 + 1j, 2 - 2j, -3 + 4j])
res = torch.angle(t)
print(res)
Результат выполнения кода:
tensor([ 0.7854, -0.7854, 2.2143])
Пример
Давайте создадим комплексный тензор из двух действительных тензоров с помощью torch.complex и вычислим фазу:
import torch
real = torch.tensor([1.0, 0.0, -1.0])
imag = torch.tensor([1.0, 1.0, 0.0])
t = torch.complex(real, imag)
res = torch.angle(t)
print(res)
Результат выполнения кода:
tensor([ 0.7854, 1.5708, 3.1416])
Пример
Вычислим фазу комплексного числа, используя отдельные действительную и мнимую части:
import torch
t = torch.complex(
torch.tensor([1.0, 2.0]),
torch.tensor([1.0, -2.0])
)
res = torch.angle(t)
print(res)
Результат выполнения кода:
tensor([ 0.7854, -0.7854])
