cupyx.scipy.sparse.linalg.cgs#
- cupyx.scipy.sparse.linalg.cgs(A, b, x0=None, tol=1e-05, maxiter=None, M=None, callback=None, atol=None)[源代码][源代码]#
使用共轭梯度平方迭代法来求解
Ax = b。- 参数:
A (ndarray, spmatrix or LinearOperator) – 线性系统的实矩阵或复矩阵，形状为
(n, n)。b (cupy.ndarray) – 线性系统的右侧，形状为
(n,)或(n, 1)。x0 (cupy.ndarray) – 解决方案的初始猜测。
tol (float) – 收敛容差。
maxiter (int) – 最大迭代次数。
M (ndarray, spmatrix or LinearOperator) –
A的预处理器。预处理器应近似于A的逆。M必须是cupy.ndarray、cupyx.scipy.sparse.spmatrix或cupyx.scipy.sparse.linalg.LinearOperator。callback (function) – 用户指定的函数，在每次迭代后调用。它被调用为
callback(xk)，其中xk是当前的解向量。atol (float) – 收敛容差。
- 返回:
它返回
x(cupy.ndarray) 和info(int)，其中x是收敛的解，而info提供收敛信息。- 返回类型: