1 files changed, 7 insertions, 2 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..4071917c2b5 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;
 
@@ -14,13 +15,17 @@ public class Matmul extends CompositeTensorFunction {
 
     private final TensorFunction argument1, argument2;
     private final String dimension;
-    
+
     public Matmul(TensorFunction argument1, TensorFunction argument2, String dimension) {
         this.argument1 = argument1;
         this.argument2 = argument2;
         this.dimension = dimension;
     }
 
+    public static TensorType outputType(TensorType a, TensorType b, String dimension) {
+        return Join.outputType(a, b);
+    }
+
     @Override
     public List<TensorFunction> functionArguments() { return ImmutableList.of(argument1, argument2); }
 
@@ -39,7 +44,7 @@ public class Matmul extends CompositeTensorFunction {
                           Reduce.Aggregator.sum,
                           dimension);
     }
-    
+
     @Override
     public String toString(ToStringContext context) {
         return "matmul(" + argument1.toString(context) + ", " + argument2.toString(context) + ", " + dimension + ")";