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)