jax.lax.rev

jax.lax.rev(operand, dimensions)

包装 XLA 的 Rev 运算符。

参数:

• operand (ArrayLike)

• dimensions (Sequence[int])

返回类型:

Array