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。