tweak accumulation of average

cvxgrp · Jan 1, 2024 · 41bf301 · 41bf301
1 parent 66bc3b2
commit 41bf301
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/aa.c b/src/aa.c
@@ -138,7 +138,9 @@ aa_float aa_apply(aa_float *f, const aa_float *x, AaWork *a) {
 
   for (i = 0; i < a->dim; ++i) {
     /* accumulate the average */
-    a->x[i] = (a->x[i] * a->iter + f[i]) / (a->iter + 1);
+    a->x[i] *=  a->iter / (a->iter + 1);
+    a->x[i] += f[i] / (a->iter + 1);
+    /* a->x[i] = (a->x[i] * a->iter + f[i]) / (a->iter + 1); */
   }
 
   a->iter++;