jax.lax.imag#

jax.lax.imag(x)[source]#

Elementwise extract imaginary part: \(\mathrm{Im}(x)\).

Returns the imaginary part of a complex number.

Parameters

x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

Return type

Array