From 2948a80d9f6dec8db0c56fab541222b250c130fc Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Tue, 3 Dec 2024 20:10:59 -0500
Subject: [PATCH] Add edm4hep::Tensor type for use in ML training/inference

---
 edm4hep.yaml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/edm4hep.yaml b/edm4hep.yaml
index 5fff0bc6..057ddd07 100644
--- a/edm4hep.yaml
+++ b/edm4hep.yaml
@@ -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"