Python PyTorch atan2用法及代碼示例

本文簡要介紹python語言中 torch.atan2 的用法。

用法:

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

參數：

• input(Tensor) -第一個輸入張量

• other(Tensor) -第二個輸入張量

關鍵字參數：

out(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 的形狀必須是可廣播的。

例子：

>>> 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])