标签： scipy pytorch sparse-matrix
我正在尝试重写一些旧代码，并从scipy切换到pytorch，并且在重写spdiags和spsolve时遇到了困难。
如何在GPU上执行这些操作？
A = spdiags(B.T,d,k,k) A = A + A.T + spdiags(D.T, 0, k, k) return spsolve(A, input)