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