jax.lax.bitwise_or#

jax.lax.bitwise_or(x,y)[source]#

Elementwise OR:\(x \vee y\).

This function lowers directly to thestablehlo.or operation.

Parameters:

• x (ArrayLike) – Input arrays. Must have matching boolean or integer dtypes.If neither is a scalar,x andy must have the same numberof dimensions and be broadcast compatible.

• y (ArrayLike) – Input arrays. Must have matching boolean or integer dtypes.If neither is a scalar,x andy must have the same numberof dimensions and be broadcast compatible.

Returns:

An array of the same dtype asx andy containing the bitwiseOR of each pair of broadcasted entries.

Return type:

Array

See also

• jax.numpy.invert(): NumPy wrapper for this API, also accessiblevia thex|y operator on JAX arrays.

• jax.lax.bitwise_not(): Elementwise NOT.

• jax.lax.bitwise_and(): Elementwise AND.

• jax.lax.bitwise_xor(): Elementwise exclusive OR.