We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7124eaa commit 6e6d24eCopy full SHA for 6e6d24e
code/numpy/transform.c
@@ -380,7 +380,7 @@ mp_obj_t transform_dot(mp_obj_t _m1, mp_obj_t _m2) {
380
if(ndim == 2) { // matrix times matrix -> matrix
381
shape = ndarray_shape_vector(0, 0, shape1, shape2);
382
} else { // matrix times vector -> vector, vector times vector -> vector (size 1)
383
- shape = ndarray_shape_vector(0, 0, 0, shape1);
+ shape = ndarray_shape_vector(0, 0, 0, shape1 * shape2);
384
}
385
ndarray_obj_t *results = ndarray_new_dense_ndarray(ndim, shape, NDARRAY_FLOAT);
386
mp_float_t *rarray = (mp_float_t *)results->array;
0 commit comments