jax.lax.rev

jax.lax.rev(operand, dimensions)

Wraps XLA’s Rev operator.

Parameters:

• operand (Union[Array, ndarray, bool_, number, bool, int, float, complex])

• dimensions (Sequence[int])

Return type:

Array