cupyx.scipy.sparse.linalg.factorized#

cupyx.scipy.sparse.linalg.factorized(A)[source]#

返回一个函数,用于求解稀疏线性系统,其中 A 已预先分解。

参数:

A (cupyx.scipy.sparse.spmatrix) – 要分解的稀疏矩阵。

返回:

一个函数,用于求解给定 A 的线性方程组。

返回类型:

可调用对象

注意

此函数使用 scipy.sparse.linalg.splu 在 CPU 上计算稀疏矩阵的 LU 分解。因此,LU 分解在 GPU 上没有加速。另一方面,使用此函数返回的方法求解线性方程的计算是在 GPU 上执行的。