jax.numpy.flipud

jax.numpy.flipud(m)

反转数组沿轴 0 的元素顺序。

JAX 实现的 numpy.flipud()。

参数:

m (ArrayLike) – 至少具有一个维度的数组。

返回值:

一个数组，其元素沿轴 0 反向排列。

返回类型:

数组

另请参阅

• jax.numpy.flip(): 反转沿给定轴的顺序

• jax.numpy.fliplr(): 反转沿轴 1 的顺序

示例

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