jax.lax.rev

jax.lax.rev(operand, dimensions)

包裝 XLA 的 Rev 運算子。

參數:

• operand (ArrayLike)

• dimensions (Sequence[int])

返回類型:

Array