Skip to content

Commit 9f0067b

Browse files
authored
add edm4eic::Tensor (#96)
This adds a new type that facilitates exchange of tensor data for ML applications.
1 parent 33e68e6 commit 9f0067b

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

edm4eic.yaml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,16 @@ components:
193193

194194
datatypes:
195195

196+
edm4eic::Tensor:
197+
Description: "Tensor type for use in training in inference of ML models"
198+
Author: "D. Kalinkin"
199+
Members:
200+
- int32_t elementType // Data type in the same encoding as "ONNXTensorElementDataType", 1 - float, 7 - int64
201+
VectorMembers:
202+
- int64_t shape // Vector of tensor lengths along its axes
203+
- float floatData // Iff elementType==1, values are stored here
204+
- int64_t int64Data // Iff elementType==7, values are stored here
205+
196206
## ==========================================================================
197207
## Particle info
198208
## ==========================================================================

0 commit comments

Comments
 (0)