jax.lax.bitwise_or#
- jax.lax.bitwise_or(x, y)[source]#
按元素或运算:\(x \vee y\)。
此函数直接降级为 stablehlo.or 操作。
- 参数:
x (ArrayLike) – 输入数组。必须具有匹配的布尔值或整数 dtype。如果两者都不是标量,则
x
和y
必须具有相同的维度数并且是广播兼容的。y (ArrayLike) – 输入数组。必须具有匹配的布尔值或整数 dtype。如果两者都不是标量,则
x
和y
必须具有相同的维度数并且是广播兼容的。
- 返回:
一个与
x
和y
具有相同 dtype 的数组,其中包含每对广播条目的按位或运算结果。- 返回类型:
另请参阅
jax.numpy.invert()
: 此 API 的 NumPy 包装器,也可以通过 JAX 数组上的x | y
运算符访问。jax.lax.bitwise_not()
: 按元素非运算。jax.lax.bitwise_and()
: 按元素与运算。jax.lax.bitwise_xor()
: 按元素异或运算。