jax.numpy.imag

jax.numpy.imag(val, /)

返回复数参数的按元素虚部。

numpy.imag 的 JAX 实现。

参数:

val (ArrayLike) – 输入数组或标量。

返回:

一个包含 val 元素虚部的数组。

返回类型:

数组

另请参阅

• jax.numpy.conjugate() 和 jax.numpy.conj()：返回输入的按元素复共轭。

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

示例

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