cupyx.scipy.sparse.linalg.splu#
- cupyx.scipy.sparse.linalg.splu(A, permc_spec=None, diag_pivot_thresh=None, relax=None, panel_size=None, options={})[source]#
计算稀疏方阵的 LU 分解。
- 参数:
A (cupyx.scipy.sparse.spmatrix) – 待分解的稀疏矩阵。
permc_spec (str) – (更多参数请参见
scipy.sparse.linalg.splu())diag_pivot_thresh (float) –
relax (int) –
panel_size (int) –
options (dict) –
- 返回:
具有
solve方法的对象。- 返回类型:
注意
此函数使用 scipy.sparse.linalg.splu 在 CPU 上对稀疏矩阵进行 LU 分解。因此，LU 分解在 GPU 上不加速。另一方面，此函数返回的对象中，使用
solve方法求解线性方程的计算是在 GPU 上执行的。