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)