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