jax.numpy.bitwise_or#

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

逐元素计算按位或（OR）操作。

JAX 对 numpy.bitwise_or 的实现。这是一个通用函数，支持在 jax.numpy.ufunc 中描述的额外 API。此函数为 JAX 数组提供了 | 运算符的实现。

参数:

  • x – 整数或布尔数组。必须能够广播到共同的形状。

  • y – 整数或布尔数组。必须能够广播到共同的形状。

  • args (ArrayLike)

  • out (None)

  • where (None)

返回:

包含逐元素按位或结果的数组。

返回类型:

任意类型

示例

显式调用 bitwise_or

>>> x = jnp.arange(4)
>>> jnp.bitwise_or(x, 1)
Array([1, 1, 3, 3], dtype=int32)

通过 | 运算符调用 bitwise_or

>>> x | 1
Array([1, 1, 3, 3], dtype=int32)