1 files changed, 4 insertions, 4 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 5e102454487..4071917c2b5 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Matmul.java
@@ -15,15 +15,15 @@ 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 Reduce.outputType(Join.outputType(a, b), ImmutableList.of(dimension));
+        return Join.outputType(a, b);
     }
 
     @Override
@@ -44,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 + ")";