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

Fix std bug.

Browse Source
This commit is contained in:
kradchen
2023-06-12 16:56:46 +08:00
parent 7ac6095912
commit 9919b09cf0
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/Function2D.cpp
View File

@@ -134,9 +134,11 @@ Matrix Aurora::std(const Matrix &aMatrix) {
int calc_size = src.getDimSize(0) == 1 ? src.getDimSize(1) : src.getDimSize(0); int calc_size = src.getDimSize(0) == 1 ? src.getDimSize(1) : src.getDimSize(0);
int col = src.getDimSize(0) == 1?1:src.getDimSize(1) ; int col = src.getDimSize(0) == 1?1:src.getDimSize(1) ;
auto std = Aurora::malloc(col); auto std = Aurora::malloc(col);
auto meanM = Aurora::mean(aMatrix);
for (int i = 0; i < col; ++i) { for (int i = 0; i < col; ++i) {
double *p = src.getData() + i * calc_size; double *p = src.getData() + i * calc_size;
double mean = cblas_dasum(calc_size, p, 1) / calc_size;
double mean = meanM[i];
vdSubI(calc_size, p, 1, &mean, 0, p, 1); vdSubI(calc_size, p, 1, &mean, 0, p, 1);
vdSqr(calc_size, p, p); vdSqr(calc_size, p, p);
std[i] = cblas_dasum(calc_size, p, 1) / (calc_size - 1); std[i] = cblas_dasum(calc_size, p, 1) / (calc_size - 1);