Save intermediate matrices.

CURENT · Mar 25, 2024 · 4f533a0 · 4f533a0
1 parent 127dc46
commit 4f533a0
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/andes/routines/eig.py b/andes/routines/eig.py
@@ -103,16 +103,18 @@ def _reduce(self, fx, fy, gx, gy, Tf, dense=True):
         spmatrix
             The reduced state matrix
         """
-        gyx = matrix(gx)
-        self.solver.linsolve(gy, gyx)
+        self.gyx = matrix(gx)
+        self.solver.linsolve(gy, self.gyx)
 
         Tfnz = Tf + np.ones_like(Tf) * np.equal(Tf, 0.0)
         iTf = spdiag((1 / Tfnz).tolist())
 
+        self.fxy = (fx - fy * self.gyx)
+
         if dense:
-            return iTf * (fx - fy * gyx)
+            return iTf * self.fxy
         else:
-            return sparse(iTf * (fx - fy * gyx))
+            return sparse(iTf * self.fxy)
 
     def _reorder(self):
         """