jax.numpy.iscomplex

jax.numpy.iscomplex(x)

返回一个布尔数组，显示输入是否为复数。

numpy.iscomplex() 的 JAX 实现。

参数:

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

返回:

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

返回类型:

Array

另请参阅

• jax.numpy.iscomplexobj()

• jax.numpy.isrealobj()

示例

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