jax.numpy.angle

jax.numpy.angle(z)[source]

Return the angle of the complex argument.

LAX-backend implementation of angle().

Original docstring below.

Parameters

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

Returns

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

..versionchanged:: 1.16.0

This function works on subclasses of ndarray like ma.array.

Return type

ndarray or scalar