root
root copied to clipboard
[tmva][sofie] Binary and Unary Operators updated for parametric tensors
This Pull Request:
Updates the implementation of Binary and Unary Operators to parse input tensors with parametric shapes, such as {batch_size1, dim1}. The method to initialize the output tensor shape can be further improved in the future.
Changes or Fixes:
- Updated implementation in the respective ROperator files to handle parametric shapes.
- Added tests as well as models for the same.
Checklist:
- [x] Tested changes locally
- [ ] Updated the documentation (if necessary)
I am working on some further changes which can be included in the PR.