1 files changed, 6 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java b/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java
index d81c02fb75f..202817ece42 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/DimensionSizes.java
@@ -71,6 +71,7 @@ public final class DimensionSizes {
      */
     public final static class Builder {
 
+        private int dimensionIndex = 0;
         private long[] sizes;
 
         public Builder(int dimensions) {
@@ -82,6 +83,11 @@ public final class DimensionSizes {
             return this;
         }
 
+        public Builder add(long size) {
+            sizes[dimensionIndex++] = size;
+            return this;
+        }
+
         /**
          * Returns the length of this in the nth dimension
          *