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 (jax.typing.ArrayLike) Return type: Array