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, 11 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 5d384e0329b..f26174d9576 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
@@ -120,6 +120,17 @@ public abstract class IndexedTensor implements Tensor {
     }
 
     @Override
+    public Double getAsDouble(TensorAddress address) {
+        try {
+            long index = toValueIndex(address, dimensionSizes, type);
+            if (index < 0 || size() <= index) return null;
+            return get(index);
+        } catch (IllegalArgumentException e) {
+            return null;
+        }
+    }
+
+    @Override
     public boolean has(TensorAddress address) {
         try {
             long index = toValueIndex(address, dimensionSizes, type);