Функция torch.atan2 в PyTorch

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

torch.atan2Это функция в PyTorch, используемая для поэлементного вычисления двумерного арктангенса (arctangent2). Она вычисляет угол на основе двух координатных значений (y, x), а возвращаемое значение выражается в радианах и находится в диапазоне [-pi, pi].

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

torch.atan2(input, other, out=None)

Описание параметра

input: Тензор числителя (координата y)
other: Тензор знаменателя (координата x)
out: Выходной тензор (необязательный)

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

Пример

import torch

# Создать тензор
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])

# Вычислить двумерный арктангенс
result = torch.atan2(y, x)
print(result)

Результат вывода:

tensor([ 0.0000,  1.5708, -1.5708,  2.3562])