cupyx.scipy.sparse.linalg.lsqr

cupyx.scipy.sparse.linalg.lsqr(A, b)

使用QR分解求解线性系统。

找到一个大型、稀疏的线性方程组的解。该函数解决了 Ax = b。给定的二维矩阵 A 被分解为 Q * R。

参数:

• A (cupy.ndarray or cupyx.scipy.sparse.csr_matrix) – 输入矩阵，维度为 (N, N)

• b (cupy.ndarray) – 右侧向量。

返回:

它的长度必须是十。它具有与SciPy相同类型的元素。只有第一个元素，即解向量 x 是可用的，其他元素表示为 None ，因为cuSOLVER的实现与SciPy的实现不同。你可以通过 norm(b - Ax) 轻松计算第四个元素，通过 norm(x) 计算第九个元素。

返回类型:

tuple

参见

scipy.sparse.linalg.lsqr()