From 9919b09cf0da3245d50846a8084666f726796919 Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Mon, 12 Jun 2023 16:56:46 +0800
Subject: [PATCH] Fix std bug.

---
 src/Function2D.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/Function2D.cpp b/src/Function2D.cpp
index d522cb9..f0a395a 100644
--- a/src/Function2D.cpp
+++ b/src/Function2D.cpp
@@ -134,9 +134,11 @@ Matrix Aurora::std(const Matrix &aMatrix) {
     int calc_size = src.getDimSize(0) == 1 ? src.getDimSize(1) : src.getDimSize(0);
     int col = src.getDimSize(0) == 1?1:src.getDimSize(1) ;
     auto std = Aurora::malloc(col);
+    auto meanM = Aurora::mean(aMatrix);
     for (int i = 0; i < col; ++i) {
         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);
         vdSqr(calc_size, p, p);
         std[i] = cblas_dasum(calc_size, p, 1) / (calc_size - 1);