jax.numpy.isreal

内容

jax.numpy.isreal#

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

返回布尔数组,显示输入在何处为实数。

JAX 实现 numpy.isreal().

参数:

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

返回值:

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

返回类型:

数组

示例

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