torch.atan2¶
- torch.atan2(input: 张量, other: 张量, *, out: Optional[张量]) 张量 ¶
逐元素的反正切,计算 并考虑象限。返回一个新张量,其中包含向量 与向量 之间的带符号角度(以弧度为单位)。(请注意,第二个参数 是 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])