import torch as th def sparse_sp2th(matrix): coo = matrix.tocoo() rows = th.from_numpy(coo.row).long().view(1, -1) cols = th.from_numpy(coo.col).long().view(1, -1) data = th.from_numpy(coo.data).float() return th.sparse.FloatTensor(th.cat((rows, cols), 0), data, coo.shape)
