jax.numpy.real

jax.numpy.real(val, /)

返回复数参数的逐元素实部。

numpy.real 的 JAX 实现。

参数:

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

返回:

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

返回类型:

Array

另请参阅

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

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

示例

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