Encode directly as float

1 files changed, 10 insertions, 1 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
index 07375cfa604..f6af1cf0ed2 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
@@ -110,7 +110,7 @@ public abstract class IndexedTensor implements Tensor {
     }
 
     /**
-     * Returns the value at the given index by direct lookup. Only use
+     * Returns the value at the given index as a double by direct lookup. Only use
      * if you know the underlying data layout.
      *
      * @param valueIndex the direct index into the underlying data.
@@ -118,6 +118,15 @@ public abstract class IndexedTensor implements Tensor {
      */
     public abstract double get(long valueIndex);
 
+    /**
+     * Returns the value at the given index as a float by direct lookup. Only use
+     * if you know the underlying data layout.
+     *
+     * @param valueIndex the direct index into the underlying data.
+     * @throws IndexOutOfBoundsException if index is out of bounds
+     */
+    public abstract float getFloat(long valueIndex);
+
     static long toValueIndex(long[] indexes, DimensionSizes sizes) {
         if (indexes.length == 1) return indexes[0]; // for speed
         if (indexes.length == 0) return 0; // for speed