jax.numpy.iscomplexobj

jax.numpy.iscomplexobj#

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

检查输入是否为复数或包含复数元素的数组。

JAX 实现 numpy.iscomplexobj().

该函数根据输入类型而不是值进行评估。具有零虚部的输入仍被视为复数。

参数::

x (Any) – 要检查的输入对象。

返回值::

如果 x 是复数或包含至少一个复数元素的数组,则为 True,否则为 False。

返回类型::

bool

示例

>>> jnp.iscomplexobj(True)
False
>>> jnp.iscomplexobj(0)
False
>>> jnp.iscomplexobj(jnp.array([1, 2]))
False
>>> jnp.iscomplexobj(1+2j)
True
>>> jnp.iscomplexobj(jnp.array([0, 1+2j]))
True