diff options
| author | Lester Solbakken <lesters@oath.com> | 2021-05-19 14:41:57 +0200 |
|---|---|---|
| committer | Lester Solbakken <lesters@oath.com> | 2021-05-19 14:41:57 +0200 |
| commit | 77807b53619a6b1449610c598f391b024eb52930 (patch) | |
| tree | 65117eab0d5e5cafee49a04b10d1c79f1e3431fa /vespajlib/src/main/java/com/yahoo/tensor | |
| parent | bd35a66573c5b6cbf05f2d875cef00817b7d23c1 (diff) | |
Revert "Revert "Add ONNX-RT evaluator to model-integration module""
This reverts commit 97080252fac0ba45b58f9d0efb56603da518428f.
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor')
| -rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java | 8 |
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) |