jax.numpy.logical_or#

jax.numpy.logical_or = <jnp.ufunc 'logical_or'>#

计算元素级的逻辑或运算。

JAX 实现的 numpy.logical_or。这是一个通用函数,并支持在 jax.numpy.ufunc 中描述的其他 API。

参数
  • x – 输入数组。必须可以广播到共同的形状。

  • y – 输入数组。必须可以广播到共同的形状。

  • args (ArrayLike)

  • out (None)

  • where (None)

返回

包含逐元素逻辑或结果的数组。

返回类型

Any

示例

>>> x = jnp.arange(4)
>>> jnp.logical_or(x, 1)
Array([ True,  True,  True,  True], dtype=bool)