torch.sparse.spsolve

torch.sparse.spsolve(input, other, *, left=True) → Tensor

计算具有唯一解的平方线性方程组的解。其目的类似于 torch.linalg.solve() ，除了该系统由具有 sparse_csr 布局的稀疏 CSR 矩阵定义。

参数:

• 输入（张量）- 表示线性系统系数的形状为（n, n）的稀疏 CSR 矩阵。

• 其他（张量）- 表示线性系统右侧的形状为（n, ）的稠密矩阵。

• 左侧（布尔值，可选）- 是否求解系统为输入 @ 输出 = 其他（默认）或输出 @ 输入 = 其他。仅支持 left=True。