jax.lax.complex

jax.lax.complex(x, y)

按元素构建复数：\(x + jy\)。

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

参数:

• x (ArrayLike) – 输入数组。必须具有匹配的浮点 dtype。如果两者都不是标量，则两个数组必须具有相同的维度数并且是广播兼容的。

• y (ArrayLike) – 输入数组。必须具有匹配的浮点 dtype。如果两者都不是标量，则两个数组必须具有相同的维度数并且是广播兼容的。

返回:

复数数组，其实部由 x 给出，虚部由 y 给出。 对于 dtype 为 float32 或 float64 的输入，结果将分别具有 dtype complex64 或 complex128。

返回类型:

数组

另请参阅

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

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

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