jax.lax.rev

目录

jax.lax.rev#

jax.lax.rev(operand, dimensions)[源代码][源代码]#

封装了 XLA 的 Rev 操作符。

参数:
  • operand (ArrayLike)

  • dimensions (Sequence[int])

返回类型:

Array