PyG 将普通矩阵(dense)转化为稀疏矩阵(edge_index, edge_weight)

PyG 将普通矩阵(dense)转化为稀疏矩阵(edge_index, edge_weight)

edge_index, edge_weight = torch_geometric.utils.dense_to_sparse(adj)

你可能感兴趣的:(笔记,pytorch)