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。