jax.numpy.imag

jax.numpy.imag(val)[source]

Return the imaginary part of the complex argument.

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