Move to iterator access where possible

1 files changed, 9 insertions, 2 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java
index 44451dc3f51..c3dce27c651 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java
@@ -4,6 +4,7 @@ package com.yahoo.tensor;
 import com.google.common.annotations.Beta;
 import com.google.common.collect.ImmutableMap;
 
+import java.util.Iterator;
 import java.util.Map;
 
 /**
@@ -26,14 +27,20 @@ public class MappedTensor implements Tensor {
 
     @Override
     public TensorType type() { return type; }
-
+    
     @Override
-    public Map<TensorAddress, Double> cells() { return cells; }
+    public int size() { return cells.size(); }
 
     @Override
     public double get(TensorAddress address) { return cells.getOrDefault(address, Double.NaN); }
 
     @Override
+    public Iterator<Map.Entry<TensorAddress, Double>> cellIterator() { return cells.entrySet().iterator(); }
+
+    @Override
+    public Map<TensorAddress, Double> cells() { return cells; }
+
+    @Override
     public int hashCode() { return cells.hashCode(); }
 
     @Override