torch.sparse.spsolve

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

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

参数

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

• other (Tensor) – 表示线性系统右端的形状为(n, )的密集矩阵。

• left (bool, 可选) – 是否求解方程为 input @ out = other（默认）或 out @ input = other。目前仅支持 left=True。