jax.numpy.fliplr

Contents

jax.numpy.fliplr#

jax.numpy.fliplr(m)[source]#

Reverse the order of elements of an array along axis 1.

JAX implementation of numpy.fliplr().

Parameters:

m (jax.typing.ArrayLike) – Array with at least two dimensions.

Returns:

An array with the elements in reverse order along axis 1.

Return type:

Array

See also

Examples

>>> x = jnp.array([[1, 2],
...                [3, 4]])
>>> jnp.fliplr(x)
Array([[2, 1],
       [4, 3]], dtype=int32)