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。