1 files changed, 6 insertions, 1 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/TensorAddress.java b/vespajlib/src/main/java/com/yahoo/tensor/TensorAddress.java
index 5c2c4d77fad..4fa759668b6 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/TensorAddress.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/TensorAddress.java
@@ -1,6 +1,7 @@
 // Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.tensor;
 
+import com.yahoo.tensor.impl.Convert;
 import com.yahoo.tensor.impl.Label;
 import com.yahoo.tensor.impl.TensorAddressAny;
 
@@ -102,7 +103,7 @@ public abstract class TensorAddress implements Comparable<TensorAddress> {
         return "'" + label + "'";
     }
 
-    /** Returns an address with only some of the dimension */
+    /** Returns an address with only some of the dimension. Ordering will also be according to indexMap */
     public TensorAddress partialCopy(int[] indexMap) {
         int[] labels = new int[indexMap.length];
         for (int i = 0; i < labels.length; ++i) {
@@ -197,6 +198,10 @@ public abstract class TensorAddress implements Comparable<TensorAddress> {
             labels[labelIndex] = Label.toNumber(label);
             return this;
         }
+
+        public Builder add(String dimension, long label) {
+            return add(dimension, Convert.safe2Int(label));
+        }
         public Builder add(String dimension, int label) {
             Objects.requireNonNull(dimension, "dimension cannot be null");
             int labelIndex = type.indexOfDimensionAsInt(dimension);