sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix matrix MxA bug

Browse Source
This commit is contained in:
kradchen
2023-12-12 15:02:34 +08:00
parent cca2358825
commit 2b49342a4d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/Matrix.cpp
View File

@@ -54,7 +54,7 @@ namespace Aurora{
aFunc(aMatrix.getDataSize(), aMatrix.getData()+1 , 2, &zero, 0, output+1 ,
2);
} else{
aFunc(aMatrix.getDataSize(), aMatrix.getData(), 1, &aScalar, 0, output, 1);
aFunc(aMatrix.getDataSize()*aMatrix.getValueType(), aMatrix.getData(), 1, &aScalar, 0, output, 1);
}
return Matrix::New(output, aMatrix.getDimSize(0), aMatrix.getDimSize(1), aMatrix.getDimSize(2),
aMatrix.getValueType());