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。