#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
csr_matrix,coo_matrix转化成torch.sparse matrix
最新推荐文章于 2024-09-04 18:10:13 发布
本文介绍了如何将Scipy库中的CSR和COO矩阵转换为PyTorch支持的稀疏矩阵格式,这对于在深度学习中处理大规模稀疏数据尤其重要。了解这种转换对于有效地在神经网络中利用稀疏数据至关重要。
摘要由CSDN通过智能技术生成