jax.numpy.isreal#

jax.numpy.isreal(x)[源]#

返回一个布尔数组,指示输入中哪些元素是实数。

JAX 对 numpy.isreal() 的实现。

参数:

x (ArrayLike) – 要检查的输入数组。

返回:

一个包含布尔值的新数组,指示实数元素。

返回类型:

Array

示例

>>> jnp.isreal(jnp.array([False, 0j, 1, 2.1, 1+2j]))
Array([ True,  True,  True,  True, False], dtype=bool)