Add faster tensor rename if dimension after rename are in the same order

1 files changed, 9 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java
index ef19ef2e96c..ec3020a1a4e 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/MappedTensor.java
@@ -42,6 +42,15 @@ public class MappedTensor implements Tensor {
     public Map<TensorAddress, Double> cells() { return cells; }
 
     @Override
+    public Tensor withType(TensorType other) {
+        if (!this.type.isRenamableTo(type)) {
+            throw new IllegalArgumentException("MappedTensor.withType: types are not compatible. Current type: '" +
+                    this.type.toString() + "', requested type: '" + type.toString() + "'");
+        }
+        return new MappedTensor(other, cells);
+    }
+
+    @Override
     public int hashCode() { return cells.hashCode(); }
 
     @Override