1 files changed, 5 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
index bb27e937699..cbb3f159623 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
@@ -3,6 +3,7 @@ package com.yahoo.tensor.functions;
 
 import com.google.common.annotations.Beta;
 import com.google.common.collect.ImmutableList;
+import com.yahoo.tensor.TensorType;
 
 import java.util.List;
 
@@ -20,6 +21,10 @@ public class Matmul extends CompositeTensorFunction {
         this.argument2 = argument2;
         this.dimension = dimension;
     }
+    
+    public static TensorType resultType(TensorType a, TensorType b, String dimension) {
+        return Reduce.resultType(Join.resultType(a, b), ImmutableList.of(dimension));
+    }
 
     @Override
     public List<TensorFunction> functionArguments() { return ImmutableList.of(argument1, argument2); }