Match order with pseudocode.

1 month ago · 710f3da157
parent dcc0916063
commit 710f3da157
1 changed files with 2 additions and 2 deletions
--- a/src/afgl/util/lanczos.py
+++ b/src/afgl/util/lanczos.py
@ -31,9 +31,9 @@ def lanczos(L, s, M):
        w = L @ V[:, j]
        alp[j] = np.dot(V[:, j], w)

-        w = w - V[:, j] * alp[j]
+        w = w - alp[j] * V[:, j]
        if j > 0:
-            w = w - V[:, j - 1] * beta[j - 1]
+            w = w - beta[j - 1] * V[:, j - 1]

        if j < M - 1:
            beta[j] = LA.norm(w)