jax.lax.imag

jax.lax.imag(x)

按元素提取虚部: \(\mathrm{Im}(x)\).

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

参数:

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

返回:

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

返回类型:

数组

另请参阅

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

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

• jax.lax.conj(): 按元素取复共轭。