jax.numpy.fliplr

内容

jax.numpy.fliplr#

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

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

JAX 实现 numpy.fliplr().

参数:

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

返回:

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

返回类型:

数组

另请参阅

示例

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