#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))
csr_matrix,coo_matrix转化成torch.sparse matrix
最新推荐文章于 2026-06-17 14:03:38 发布
本文介绍了如何将Scipy库中的CSR和COO矩阵转换为PyTorch支持的稀疏矩阵格式,这对于在深度学习中处理大规模稀疏数据尤其重要。了解这种转换对于有效地在神经网络中利用稀疏数据至关重要。

707

被折叠的 条评论
为什么被折叠?



