jax.numpy.flip

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

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

JAX 实现的 numpy.flip()。

参数:

• m (ArrayLike) – 数组

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

返回:

沿 axis 轴反向排列的元素数组。

返回类型:

Array

参见

• 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)