Skip to content
Snippets Groups Projects

feat: Backward of Stack Operator

Merged Jerome Hue requested to merge jeromeh/aidge_core:stack-backward into dev
1 file
+ 0
3
Compare changes
  • Side-by-side
  • Inline
+ 0
3
@@ -68,9 +68,6 @@ void StackOpImpl::backward() {
auto inputGrad = op.getInput(0)->grad();
auto outputGrad = op.getOutput(0)->grad();
Log::notice("Size of stack in grad : {}", inputGrad->size());
Log::notice("Size of stack out grad : {}", outputGrad->size());
*inputGrad = outputGrad->extract({op.backwardStep() -1 }).clone();
}
Loading