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)[source]#
使用共轭梯度平方迭代求解
Ax = b。- 参数:
A (ndarray, spmatrix 或 LinearOperator) – 线性系统的实数或复数矩阵，形状为
(n, n)。b (cupy.ndarray) – 线性系统的右手边，形状为
(n,)或(n, 1)。x0 (cupy.ndarray) – 解决方案的起始猜测。
tol (float) – 收敛容差。
maxiter (int) – 最大迭代次数。
M (ndarray, spmatrix 或 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提供收敛信息。- 返回类型: