Skip to content

feat: Backward of Stack Operator

Jerome Hue requested to merge jeromeh/aidge_core:stack-backward into dev

Added the backward pass implementation for the Stack operator. The backward pass extract gradients from the stacked output grad.

Key changes:

  • Implemented StackOpImpl::backward() method to handle gradient propagation
  • Added BackwardStep counter attribute to track backward pass position
  • Extended unit tests to verify gradient computation
Edited by Jerome Hue

Merge request reports

Loading