jax.lax.conj

jax.lax.conj(x)

逐元素复共轭函数: \(\overline{x}\)。

此函数降低为 stablehlo.real、stablehlo.imag 和 stablehlo.complex 的组合。

参数:

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

返回:

与 x 形状和 dtype 相同的数组，包含其复共轭。

返回类型:

Array

另请参阅

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

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

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

• jax.lax.abs(): 逐元素绝对值 / 复数模。