jax.numpy.angle

jax.numpy.angle(z, deg=False)

Return the angle of the complex argument.

LAX-backend implementation of numpy.angle().

Original docstring below.

Parameters:

• z (array_like) – A complex number or sequence of complex numbers.

• deg (bool, optional) – Return angle in degrees if True, radians if False (default).

Returns:

angle – The counterclockwise angle from the positive real axis on the complex plane in the range (-pi, pi], with dtype as numpy.float64.

Changed in version 1.16.0: This function works on subclasses of ndarray like ma.array.

Return type:

ndarray or scalar