Merge pull request #30000 from vespa-engine/balder/add-method-returning-possible-object-to-avoid-double-lookupv8.290.51

Balder/add method returning possible object to avoid double lookup

1 files changed, 10 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
index cf6e737bf27..aab4a4e1297 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
@@ -83,6 +83,16 @@ public class MixedTensor implements Tensor {
     }
 
     @Override
+    public Double getAsDouble(TensorAddress address) {
+        var block = index.blockOf(address);
+        int denseOffset = index.denseOffsetOf(address);
+        if (block == null || denseOffset < 0 || denseOffset >= block.cells.length) {
+            return null;
+        }
+        return block.cells[denseOffset];
+    }
+
+    @Override
     public boolean has(TensorAddress address) {
         var block = index.blockOf(address);
         int denseOffset = index.denseOffsetOf(address);