jax.numpy.imag

jax.numpy.imag(val, /)

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

JAX 对 numpy.imag 的实现。

参数:

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

返回:

包含 val 元素虚部的数组。

返回类型:

Array

另请参阅

• 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)