Fix sum bug

This commit is contained in:
Krad
2023-04-25 17:46:25 +08:00
parent c71e359ce5
commit d1389df8a4

View File

@@ -345,7 +345,7 @@ Matrix Aurora::sum(const Matrix &aMatrix, FunctionDirection direction) {
{ {
Eigen::Map<Eigen::MatrixXd> srcM(aMatrix.getData(),aMatrix.getDimSize(0),aMatrix.getDimSize(1)); Eigen::Map<Eigen::MatrixXd> srcM(aMatrix.getData(),aMatrix.getDimSize(0),aMatrix.getDimSize(1));
double * ret = malloc(aMatrix.getDimSize(1)); double * ret = malloc(aMatrix.getDimSize(1));
Eigen::Map<Eigen::VectorXd> retV(ret,aMatrix.getDimSize(0)); Eigen::Map<Eigen::VectorXd> retV(ret,aMatrix.getDimSize(1));
retV = srcM.colwise().sum(); retV = srcM.colwise().sum();
return Matrix::New(ret,1,aMatrix.getDimSize(1)); return Matrix::New(ret,1,aMatrix.getDimSize(1));
} }