jax.lax.rev# jax.lax.rev(operand, dimensions)[source]# Wraps XLA’s Rev operator. Parameters: operand (jax.typing.ArrayLike) dimensions (Sequence[int]) Return type: Array
