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。