Merge pull request #37 from ChabaneAmaury/forward_vectorization

Improved f solver calculation
eitcom · Apr 9, 2022 · a40f85a · a40f85a
2 parents 9279ba2 + 5428272
commit a40f85a
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/pyeit/eit/fem.py b/pyeit/eit/fem.py
@@ -238,10 +238,7 @@ def solve_nd(self, ex_mat, perm):
         # 4. solving nodes potential using boundary conditions
         b = self._natural_boundary_nd(ex_mat)
 
-        def f_init(b):
-            return np.dot(r_matrix, b).ravel()
-
-        f = np.array(list(map(f_init, b)))
+        f = np.dot(r_matrix, b[:, None]).T.reshape(b.shape[:-1])
 
         # 5. build Jacobian matrix column wise (element wise)
         #    Je = Re*Ke*Ve = (nex3) * (3x3) * (3x1)