jax.numpy.flip

jax.numpy.flip(m, axis=None)

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

JAX 实现的 numpy.flip()。

参数:

• m (ArrayLike) – 数组。

• axis (int | Sequence[int] | None | None) – 整数或整数序列。指定应该沿哪个或哪些轴反转数组元素。默认为 None，表示沿所有轴反转。

返回:

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

返回类型:

数组

另请参阅

• jax.numpy.fliplr(): 沿轴 1 反转顺序（左右）

• jax.numpy.flipud(): 沿轴 0 反转顺序（上下）

示例

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

如果 axis 指定为整数，则 jax.numpy.flip 仅沿该特定轴反转数组。

>>> jnp.flip(x1, axis=1)
Array([[2, 1],
       [4, 3]], dtype=int32)

>>> x2 = jnp.arange(1, 9).reshape(2, 2, 2)
>>> x2
Array([[[1, 2],
        [3, 4]],

       [[5, 6],
        [7, 8]]], dtype=int32)
>>> jnp.flip(x2)
Array([[[8, 7],
        [6, 5]],

       [[4, 3],
        [2, 1]]], dtype=int32)

当 axis 指定为整数序列时，jax.numpy.flip 沿指定的轴反转数组。

>>> jnp.flip(x2, axis=[1, 2])
Array([[[4, 3],
        [2, 1]],

       [[8, 7],
        [6, 5]]], dtype=int32)