jaxopt.linear_solve.solve_gmres
- jaxopt.linear_solve.solve_gmres(matvec, b, ridge=None, init=None, tol=1e-05, **kwargs)[source]
Solves
A x = b
using gmres.- Parameters
matvec (
Callable
) – product betweenA
and a vector.b (
Any
) – pytree.ridge (
Optional
[float
]) – optional ridge regularization.init (
Optional
[Any
]) – optional initialization to be used by gmres.**kwargs – additional keyword arguments for solver.
tol (float) –
- Return type
Any
- Returns
pytree with same structure as
b
.