jax.scipy.sparse.linalg.bicgstab#
- jax.scipy.sparse.linalg.bicgstab(A, b, x0=None, *, tol=1e-05, atol=0.0, maxiter=None, M=None)[source]#
使用双共轭梯度稳定迭代求解
Ax = b。JAX 的
bicgstab的数值结果应该与 SciPy 的bicgstab完全匹配（在数值精度范围内），但请注意接口略有不同：您需要提供线性算子A作为函数，而不是稀疏矩阵或LinearOperator。与
cg一样，bicgstab的导数是通过隐式微分实现的，使用另一个bicgstab求解，而不是通过求解器进行微分。只有当两个求解都收敛时，它们才是准确的。- 参数：
A (ndarray, function 或 支持矩阵乘法的对象) – 二维数组或函数，在像
A(x)或A @ x那样调用时计算线性映射（矩阵向量积）Ax。A可以表示任何一般的（非对称）线性算子，函数必须返回与其参数具有相同结构和形状的数组。b (array 或 tree of arrays) – 线性方程组的右端，表示单个向量。可以存储为数组或 Python 容器中的数组，具有任意形状。
x0 (array 或 tree of arrays) – 解决方案的初始猜测。必须与
b具有相同的结构。tol (float, 可选) – 收敛容差，
norm(residual) <= max(tol*norm(b), atol)。我们没有实现 SciPy 的“传统”行为，因此除非您明确地将atol传递给 SciPy 的cg，否则 JAX 的容差将与 SciPy 不同。atol (float, 可选) – 收敛容差，
norm(residual) <= max(tol*norm(b), atol)。我们没有实现 SciPy 的“传统”行为，因此除非您明确地将atol传递给 SciPy 的cg，否则 JAX 的容差将与 SciPy 不同。maxiter (integer) – 最大迭代次数。即使未达到指定的容差，迭代也会在 maxiter 步后停止。
M (ndarray, function 或 支持矩阵乘法的对象) – A 的预处理器。预处理器应该近似于 A 的逆。有效的预处理会显著提高收敛速度，这意味着达到给定误差容差所需的迭代次数更少。
- 返回：
x (array 或 tree of arrays) – 收敛的解。与
b具有相同的结构。info (None) – 收敛信息的占位符。将来，JAX 将报告未达到收敛时的迭代次数，就像 SciPy 一样。