jax.numpy.iscomplex

jax.numpy.iscomplex(x)

返回一个布尔数组，显示输入中哪些元素是复数。

JAX 实现的 numpy.iscomplex()。

参数:

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

返回:

一个新数组，其中包含指示复数元素的布尔值。

返回类型:

数组

另请参阅

• jax.numpy.iscomplexobj()

• jax.numpy.isrealobj()

示例

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