jax.numpy.imag

jax.numpy.imag(val, /)

Return the imaginary part of the complex argument.

LAX-backend implementation of numpy.imag().

Original docstring below.

Parameters:

val (array_like) – Input array.

Returns:

out – The imaginary component of the complex argument. If val is real, the type of val is used for the output. If val has complex elements, the returned type is float.

Return type:

ndarray or scalar