jax.lax.rev#

jax.lax.rev(operand, dimensions)[source]#

Wraps XLA’s Rev operator.

Parameters
Return type

Array