feat: Backward of Stack Operator
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
Activity
Filter activity
added 1 commit
- 8e8606cc - feat: Implement backward() for stack, add unit test
added 55 commits
-
8e8606cc...03de25a0 - 52 commits from branch
eclipse/aidge:dev
- dba5e80d - Fix StackOp forward()
- 5e406854 - Add backward step counter and backward() declaration for stack op
- 01c74a6a - feat: Implement backward() for stack, add unit test
Toggle commit list-
8e8606cc...03de25a0 - 52 commits from branch
requested review from @olivierbichler
enabled an automatic merge when all merge checks for 0c6e3302 pass
mentioned in commit 3ee75f49
changed milestone to %aidge v0.6.0
added Feature 🚀 TopicOperator labels
assigned to @jeromeh
mentioned in merge request !345 (closed)
Please register or sign in to reply