【scipy.sparse矩阵转torch.sparse】稀疏矩阵pytorch的转换

values =X_train.data
indices = np.vstack((X_train.row, X_train.col))
i = torch.LongTensor(indices)
v = torch.FloatTensor(values)
shape = X_train.shape
X_train=torch.sparse.FloatTensor(i, v, torch.Size(shape))

注:
转换为scipy.sparse:.tocoo()即可

上一篇:Git使用之稀疏检出


下一篇:Sparse Table 算法模板