jax.lax.imag

jax.lax.imag(x)

逐元素提取虚部：\(\mathrm{Im}(x)\)。

此函数直接降低为 stablehlo.imag 操作。

参数:

x (ArrayLike) – 输入数组。必须具有复数 dtype。

返回:

与 x 形状相同的数组，包含其虚部。如果 x.dtype == complex64，则 dtype 将为 float32；如果 x.dtype == complex128，则 dtype 将为 float64。

返回类型:

数组

另请参阅

• jax.lax.complex()：逐元素构造复数。

• jax.lax.real()：逐元素提取实部。

• jax.lax.conj()：逐元素复共轭。