jax.numpy.bitwise_and — JAX documentation (original) (raw)

jax.numpy.bitwise_and#

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

Compute the bitwise AND operation elementwise.

JAX implementation of numpy.bitwise_and. This is a universal function, and supports the additional APIs described at jax.numpy.ufunc. This function provides the implementation of the & operator for JAX arrays.

Parameters:

• x – integer or boolean arrays. Must be broadcastable to a common shape.
• y – integer or boolean arrays. Must be broadcastable to a common shape.
• args (ArrayLike)
• out (None)
• where (None)

Returns:

Array containing the result of the element-wise bitwise AND.

Return type:

Any

Examples

Calling bitwise_and explicitly:

x = jnp.arange(4) jnp.bitwise_and(x, 1) Array([0, 1, 0, 1], dtype=int32)

Calling bitwise_and via the & operator:

x & 1 Array([0, 1, 0, 1], dtype=int32)