jax.lax.reduce_or#

jax.lax.reduce_or(operand,axes)[source]#

Compute the bitwise OR of elements over one or more array axes.

Parameters:

• operand (ArrayLike) – array over which to compute the reduction. Must have booleanor integer dtype.

• axes (Sequence[int]) – sequence of zero or more unique integers specifying the axes overwhich to reduce. Each entry must satisfy0<=axis<operand.ndim.

Returns:

An array of the same dtype asoperand, with shape correspondingto the dimensions ofoperand.shape withaxes removed.

Return type:

Array

See also

• jax.numpy.bitwise_or.reduce(): more flexible NumPy-style logicalreduction API, built aroundjax.lax.reduce_or().

• Other low-leveljax.lax reduction operators:jax.lax.reduce_sum(),jax.lax.reduce_prod(),jax.lax.reduce_max(),jax.lax.reduce_min(),jax.lax.reduce_and(),jax.lax.reduce_xor().