jax.numpy.arctan2

jax.numpy.arctan2(x1, x2)

Element-wise arc tangent of x1/x2 choosing the quadrant correctly.

LAX-backend implementation of arctan2().

Original docstring below.

The quadrant (i.e., branch) is chosen so that arctan2(x1, x2) is the signed angle in radians between the ray ending at the origin and passing through the point (1,0), and the ray ending at the origin and passing through the point (x2, x1). (Note the role reversal: the “y-coordinate” is the first function parameter, the “x-coordinate” is the second.) By IEEE convention, this function is defined for x2 = +/-0 and for either or both of x1 and x2 = +/-inf (see Notes for specific values).

This function is not defined for complex-valued arguments; for the so-called argument of complex values, use angle.

Parameters
  • x1 (array_like, real-valued) – y-coordinates.

  • x2 (array_like, real-valued) – x-coordinates. If x1.shape != x2.shape, they must be broadcastable to a common shape (which becomes the shape of the output).

Returns

angle – Array of angles in radians, in the range [-pi, pi]. This is a scalar if both x1 and x2 are scalars.

Return type

ndarray

References

1

ISO/IEC standard 9899:1999, “Programming language C.”