pylops_gpu.optimization.cg.cg¶
-
pylops_gpu.optimization.cg.
cg
(A, y, x=None, niter=10, tol=1e-10)[source]¶ Conjugate gradient
Solve a system of equations given the square operator
A
and datay
using conjugate gradient iterations.Parameters: - A :
pylops_gpu.LinearOperator
Operator to invert of size \([N \times N]\)
- y :
torch.Tensor
Data of size \([N \times 1]\)
- x0 :
torch.Tensor
, optional Initial guess
- niter :
int
, optional Number of iterations
- tol :
int
, optional Residual norm tolerance
Returns: - x :
torch.Tensor
Estimated model
- iiter :
torch.Tensor
Max number of iterations model
- A :