Module burn::backend::autodiff::checkpoint::retro_forward

Expand description

RetroForward module

Traitsยง

  • Definition of the forward function of a node, called during retropropagation only. This is different from the normal forward function because it reads and writes from the BackwardStates map instead of having a clear function signature.