jax.numpy.conjugate

jax.numpy.conjugate(x, /)

返回输入的逐元素复共轭。

numpy.conjugate 的 JAX 实现。

参数:

x (类数组) – 输入数组或标量。

返回:

一个包含 x 的复共轭的数组。

返回类型:

数组

另请参阅

• jax.numpy.real(): 返回复数参数的逐元素实部。

• jax.numpy.imag(): 返回复数参数的逐元素虚部。

示例

>>> jnp.conjugate(3)
Array(3, dtype=int32, weak_type=True)
>>> x = jnp.array([2-1j, 3+5j, 7])
>>> jnp.conjugate(x)
Array([2.+1.j, 3.-5.j, 7.-0.j], dtype=complex64)