jax.numpy.flipud

jax.numpy.flipud(m)

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

numpy.flipud() 的 JAX 实现。

参数:

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

返回:

一个沿轴 0 元素顺序相反的数组。

返回类型:

数组 (Array)

参见

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

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

示例

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