jax.lax.real

jax.lax.real(x)

逐元素提取实部：\(\mathrm{Re}(x)\)。

此函数直接降低到 stablehlo.real 操作。

参数:

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

返回:

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

返回类型:

数组

另请参阅

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

• jax.lax.imag()：逐元素提取虚部。

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