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)