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.