jax.numpy.flipud

内容

jax.numpy.flipud#

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

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

JAX 实现 numpy.flipud().

参数:

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

返回值:

沿轴 0 反转顺序后的数组。

返回类型:

数组

参见

示例

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