1 files changed, 9 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java
index cfc78be7e0c..aa28a26deb2 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java
@@ -61,6 +61,15 @@ public class Reduce extends PrimitiveTensorFunction {
         this.dimensions = ImmutableList.copyOf(dimensions);
     }
 
+    public static TensorType resultType(TensorType type, List<String> reduceDimensions) {
+        TensorType.Builder b = new TensorType.Builder();
+        for (TensorType.Dimension dimension : type.dimensions()) {
+            if ( ! reduceDimensions.contains(dimension.name()))
+                b.dimension(dimension);
+        }
+        return b.build();
+    }
+    
     public TensorFunction argument() { return argument; }
 
     @Override