root icon indicating copy to clipboard operation
root copied to clipboard

[tmva][sofie] Binary and Unary Operators updated for parametric tensors

Open extint opened this issue 1 year ago • 1 comments

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)

extint avatar Jul 10 '24 22:07 extint

I am working on some further changes which can be included in the PR.

extint avatar Jul 23 '24 09:07 extint