Skip to content
Snippets Groups Projects

feat: Backward of Stack Operator

Merged 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
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply
Loading