cupyx.scipy.sparse.linalg.lsqr#
- cupyx.scipy.sparse.linalg.lsqr(A, b)[source]#
使用 QR 分解求解线性系统。
求解大型稀疏线性方程组。该函数求解
Ax = b
。给定二维矩阵A
被分解为Q * R
。- 参数:
A (cupy.ndarray 或 cupyx.scipy.sparse.csr_matrix) – 维度为
(N, N)
的输入矩阵b (cupy.ndarray) – 右侧向量。
- 返回:
其长度必须为十。它的元素类型与 SciPy 相同。由于 cuSOLVER 的实现与 SciPy 不同,只有第一个元素(解向量
x
)可用,其他元素表示为None
。您可以通过norm(b - Ax)
轻松计算第四个元素,并通过norm(x)
计算第九个元素。- 返回类型: