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。

返回类型:

Array

参见

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

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

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