Функция PyTorch torch.arctan2

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

torch.arctan2является функцией в PyTorch, используемой для поэлементного вычисления арктангенса двух аргументов (arctangent2), с теми жеtorch.atan2функциональность.

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

torch.arctan2(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.arctan2(y, x)
print(result)

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

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