#adjM is csr_matrix
coo_adj=adjM.tocoo()
values = coo_adj.data
indices = np.vstack((coo_adj.row, coo_adj.col))
i = torch.LongTensor(indices)
v = torch.FloatTensor(values)
shape = coo_adj.shape
#torch_sparse_adj is torch.sparse matrix
torch_sparse_adj=torch.sparse.FloatTensor(i, v, torch.Size(shape))
09-03
969
04-06
2316
09-28
1万+
09-29
3307
12-01
2619
01-11
1143
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交