jax.lax.rev

jax.lax.rev(operand, dimensions)

封装 XLA 的 Rev 运算符。

参数：

• operand (ArrayLike)

• dimensions (Sequence[int])

返回类型：

数组