jax.numpy.flipud

Contents

jax.numpy.flipud#

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

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

JAX implementation of numpy.flipud().

Parameters:

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

Returns:

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

Return type:

Array

See also

Examples

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