summarized computations in krylov sampling

478a37dd · Reimar H Leike · adc1eca5 · 478a37dd
Commit 478a37dd authored Apr 24, 2018 by Reimar H Leike
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

nifty4/library/krylov_sampling.py nifty4/library/krylov_sampling.py +1 -2

No files found.
--- a/nifty4/library/krylov_sampling.py
+++ b/nifty4/library/krylov_sampling.py
@@ -65,8 +65,7 @@ def generate_krylov_samples(D_inv, S, j=None,  N_samps=1, N_iter=10,
            break
        x += gamma*p
        for i in range(N_samps):
-            y[i] -= p.vdot(D_inv(y[i])) * p / d
-            y[i] += randn() / sqrt(d) * p
+            y[i] += (randn() * sqrt(d) - p.vdot(D_inv(y[i]))) / d * p
        r_new = r - gamma * D_inv(p)
        beta = r_new.vdot(r_new) / r.vdot(r)
        r = r_new