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