torch.atan2¶
- torch.atan2(input, other, *, out=None) Tensor ¶
Element-wise arctangent of with consideration of the quadrant. Returns a new tensor with the signed angles in radians between vector and vector . (Note that , the second parameter, is the x-coordinate, while , the first parameter, is the y-coordinate.)
The shapes of
input
andother
must be broadcastable.- Parameters
- Keyword Arguments
out (Tensor, optional) – the output tensor.
Example:
>>> 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])