Add ONNX-RT evaluator to model-integration module

1 files changed, 8 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
index 24578f035a3..d822a5c6b8b 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
@@ -191,6 +191,14 @@ public abstract class IndexedTensor implements Tensor {
 
     public DimensionSizes dimensionSizes() { return dimensionSizes; }
 
+    public long[] shape() {
+        long[] result = new long[dimensionSizes.dimensions()];
+        for (int i = 0; i < result.length; ++i) {
+            result[i] = dimensionSizes.size(i);
+        }
+        return result;
+    }
+
     @Override
     public Map<TensorAddress, Double> cells() {
         if (dimensionSizes.dimensions() == 0)