jax.lax.bitwise_xor

jax.lax.bitwise_xor(x, y)

Elementwise exclusive OR: \(x \oplus y\).

This function lowers directly to the stablehlo.xor operation.

Parameters:

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

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

Returns:

An array of the same dtype as x and y containing the bitwise XOR of each pair of broadcasted entries.

Return type:

Array

See also

• jax.numpy.bitwise_xor(): NumPy wrapper for this API, also accessible via the x ^ y operator on JAX arrays.

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

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

• jax.lax.bitwise_or(): Elementwise OR.