jax.numpy.iscomplex

jax.numpy.iscomplex(x)

返回布林陣列，顯示輸入是否為複數。

JAX 版本的 numpy.iscomplex() 實作。

參數:

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)