Add float accessor

1 files changed, 10 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 f6af1cf0ed2..d43e9ee74a3 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
@@ -97,6 +97,16 @@ public abstract class IndexedTensor implements Tensor {
         return get((int)toValueIndex(indexes, dimensionSizes));
     }
 
+    /**
+     * Returns the value at the given indexes as a float
+     *
+     * @param indexes the indexes into the dimensions of this. Must be one number per dimension of this
+     * @throws IndexOutOfBoundsException if any of the indexes are out of bound or a wrong number of indexes are given
+     */
+    public float getFloat(long ... indexes) {
+        return getFloat((int)toValueIndex(indexes, dimensionSizes));
+    }
+
     /** Returns the value at this address, or NaN if there is no value at this address */
     @Override
     public double get(TensorAddress address) {