jax.lax.rev

jax.lax.rev(operand, dimensions)

Wraps XLA’s Rev operator.

Parameters

• operand (Any) –

• dimensions (Sequence[int]) –

Return type

Any