Ctrl+K

jax.numpy.atan2

Contents

jax.numpy.atan2#

jax.numpy.atan2(x1, x2, /)[source]#

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

LAX-backend implementation of numpy.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

Contents