torch.atan2¶
- torch.atan2(input, other, *, out=None) Tensor ¶
的逐元素反正切,考虑象限。返回一个新张量,其中包含向量 和向量 之间的有符号弧度。 (注意,,第二个参数,是 x 坐标,而 ,第一个参数,是 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])