Rate this Page

torch.atan2#

torch.atan2(input:Tensor,other:Tensor,*,out:Optional[Tensor])Tensor#

Element-wise arctangent ofinputi/otheri\text{input}_{i} / \text{other}_{i}with consideration of the quadrant. Returns a new tensor with the signed anglesin radians between vector(otheri,inputi)(\text{other}_{i}, \text{input}_{i})and vector(1,0)(1, 0). (Note thatotheri\text{other}_{i}, the secondparameter, is the x-coordinate, whileinputi\text{input}_{i}, the firstparameter, is the y-coordinate.)

The shapes ofinput andother must bebroadcastable.

Parameters
  • input (Tensor) – the first input tensor

  • other (Tensor) – the second input tensor

Keyword Arguments

out (Tensor,optional) – the output tensor.

Example:

>>>a=torch.randn(4)>>>atensor([ 0.9041,  0.0196, -0.3108, -2.4423])>>>torch.atan2(a,torch.randn(4))tensor([ 0.9833,  0.0811, -1.9743, -1.4151])