torch.atan2

torch.atan2(input: Tensor, other: Tensor, *, out: Tensor | None) → Tensor

Поэлементный арктангенс $\text{input}_{i} / \text{other}_{i}$ с учётом квадранта. Возвращает новый тензор со знаковыми углами в радианах между вектором $(\text{other}_{i}, \text{input}_{i})$ и вектором $(1, 0)$. (Обратите внимание: $\text{other}_{i}$, второй параметр, является координатой x, а $\text{input}_{i}$, первый параметр, — координатой y.)

Формы input и other должны быть совместимы для широковещательной передачи.

Параметры:

• input (Tensor) – первый входной тензор
• other (Tensor) – второй входной тензор

Именованные аргументы:

out (Tensor, необязательный) – выходной тензор.

Пример:

>>> a = torch.randn(4)
>>> a
tensor([ 0.9041,  0.0196, -0.3108, -2.4423])
>>> torch.atan2(a, torch.randn(4))
tensor([ 0.9833,  0.0811, -1.9743, -1.4151])