Skip to content

Commit

Permalink
Add edm4hep::Tensor type for use in ML training/inference
Browse files Browse the repository at this point in the history
  • Loading branch information
veprbl committed Dec 4, 2024
1 parent fe5a540 commit 2948a80
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions edm4hep.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -821,6 +821,16 @@ datatypes:
- float amplitude // calibrated detector data


edm4hep::Tensor:
Description: "Tensor type for use in training/inference for ML models"
Author: "EDM4hep authors"
Members:
- int32_t elementType // Data type in the same encoding as "ONNXTensorElementDataType", 1 - float, 7 - int64
VectorMembers:
- int64_t shape // Vector of tensor lengths along its axes
- float floatData // Iff elementType==1, values are stored here
- int64_t int64Data // Iff elementType==7, values are stored here


edm4hep::RecDqdx:
Description: "dN/dx or dE/dx info of a Track"
Expand Down

0 comments on commit 2948a80

Please sign in to comment.