jax.numpy.logical_or#

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

逐元素计算逻辑 OR 运算。

(NumPy v2.2 中) numpy.logical_or 的 JAX 实现。这是一个通用函数，并支持 jax.numpy.ufunc 中描述的附加 API。

参数:

  • x – 输入数组。必须可广播到公共形状。

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

  • args (ArrayLike)

  • out (None)

  • where (None)

返回值:

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

返回类型:

Any

示例

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