Pad dimensions of matmul argument to handle vector case.

almostrealism · Sep 28, 2024 · b4f0399 · b4f0399
1 parent 98a1733
commit b4f0399
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algebra/src/main/java/org/almostrealism/algebra/MatrixFeatures.java b/algebra/src/main/java/org/almostrealism/algebra/MatrixFeatures.java
@@ -57,7 +57,7 @@ default <T extends PackedCollection<?>> CollectionProducer<T> matmul(Producer<T>
 		int n = shape.length(1);
 
 		if (enableCollectionExpression) {
-			int p = vshape.length(1);
+			int p = padDimensions(vshape, 1, 2).length(1);
 
 			TraversalPolicy resultShape = shape(m, p);