scipy.sparse.linalg.
bicgstab#
- scipy.sparse.linalg.bicgstab(A, b, x0=None, *, rtol=1e-05, atol=0.0, maxiter=None, M=None, callback=None)[原始碼]#
使用雙共軛梯度穩定迭代法來解
Ax = b
。- 參數:
- A{稀疏陣列, ndarray, LinearOperator}
線性系統的實數或複數 N 乘 N 矩陣。或者,A 可以是一個線性運算子,可以使用例如
scipy.sparse.linalg.LinearOperator
來產生Ax
和A^T x
。- bndarray
線性系統的右手邊。形狀為 (N,) 或 (N,1)。
- x0ndarray
解的起始猜測值。
- rtol, atolfloat, 選用
收斂性測試的參數。為了收斂,應該滿足
norm(b - A @ x) <= max(rtol*norm(b), atol)
。預設值為atol=0.
和rtol=1e-5
。- maxiterinteger
最大迭代次數。即使未達到指定的容忍度,迭代也會在 maxiter 步驟後停止。
- M{稀疏陣列, ndarray, LinearOperator}
A 的預處理器。它應該近似於 A 的逆矩陣(請參閱註釋)。有效的預處理可以顯著提高收斂速度,這表示達到給定的誤差容忍度所需的迭代次數更少。
- callbackfunction
使用者提供的函數,在每次迭代後呼叫。它被呼叫為
callback(xk)
,其中xk
是目前的解向量。
- 回傳值:
- xndarray
已收斂的解。
- infointeger
- 提供收斂資訊
0 : 成功退出 >0 : 未達到容忍度而收斂,迭代次數 <0 : 參數崩潰
註釋
預處理器 M 應該是一個矩陣,使得
M @ A
的條件數小於 A,請參閱 [1] 。參考文獻
[1]“預處理器”,維基百科,https://en.wikipedia.org/wiki/Preconditioner
[2]“雙共軛梯度穩定法”,維基百科,https://en.wikipedia.org/wiki/Biconjugate_gradient_stabilized_method
範例
>>> import numpy as np >>> from scipy.sparse import csc_array >>> from scipy.sparse.linalg import bicgstab >>> R = np.array([[4, 2, 0, 1], ... [3, 0, 0, 2], ... [0, 1, 1, 1], ... [0, 2, 1, 0]]) >>> A = csc_array(R) >>> b = np.array([-1, -0.5, -1, 2]) >>> x, exit_code = bicgstab(A, b, atol=1e-5) >>> print(exit_code) # 0 indicates successful convergence 0 >>> np.allclose(A.dot(x), b) True