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 (Any) –

Return type

Any