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, optional) – 是否求解系统 input @ out = other（默认）或 out @ input = other。仅支持 left=True。