torch.sparse.spsolve

torch.sparse.spsolve(input, other, *, left=True) → Tensor

计算具有唯一解的线性方程组的平方系统的解。其目的是类似于 torch.linalg.solve()，除了该系统是由布局为 sparse_csr 的稀疏 CSR 矩阵定义的。

参数

• input (Tensor) – 形状为 (n, n) 的稀疏 CSR 矩阵，表示线性系统的系数。

• other (Tensor) – 形状为 (n, ) 的稠密矩阵，表示线性方程组的右侧。

• left (bool, 可选) – 是否求解 input @ out = other (默认) 或 out @ input = other 的系统。仅支持 left=True。