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++;