def solve(self, A, b): from pyamg.relaxation import smoothing from pyamg import multilevel_solver lvl = multilevel_solver.level() lvl.A = A fn = getattr(smoothing, 'setup_' + str(solver)) relax = fn(lvl, **kwargs) x = np.zeros_like(b) relax(A, x, b) return x
def solve(self, A, b): from pyamg.relaxation import smoothing from pyamg import multilevel_solver lvl = multilevel_solver.level() lvl.A = A fn = getattr(smoothing, 'setup_' + str(solver)) relax = fn(lvl, **kwargs) x = np.zeros_like(b) relax(A, x, b) return x