Fix consistency of optimized dot

numpy · Mar 13, 2006 · 55b8604 · 55b8604
1 parent aa04352
commit 55b8604
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/numpy/core/blasdot/_dotblas.c b/numpy/core/blasdot/_dotblas.c
@@ -248,8 +248,13 @@ dotblas_matrixproduct(PyObject *dummy, PyObject *args)
 	    }
 	    l = dimensions[0];
 	}
-	else {
-	    nd = ap1->nd;
+	else if (ap1shape == _scalar) {
+            nd = 0;
+            l = 1;
+            dimensions[0] = 1;
+        }
+        else {
+            nd = ap1->nd;
 	    for (l = 1, j = 0; j < nd; j++) {
 		dimensions[j] = ap1->dimensions[j];
 		l *= dimensions[j];