jax.lax.imag

jax.lax.imag(x)

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

Returns the imaginary part of a complex number.

Parameters:

x (ArrayLike)

Return type:

Array