jax.numpy.flipud

jax.numpy.flipud(m)

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

JAX implementation of numpy.flipud().

Parameters:

m (ArrayLike) – Array with at least one dimension.

Returns:

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

Return type:

Array

See also

• jax.numpy.flip(): reverse the order along the given axis

• jax.numpy.fliplr(): reverse the order along axis 1

Examples

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