Pytorch几何稀疏邻接矩阵到边缘索引张量

【问题标题】：Pytorch Geometric sparse adjacency matrix to edge index tensorPytorch几何稀疏邻接矩阵到边缘索引张量
【发布时间】：2021-09-07 15:39:53
【问题描述】：

我的数据对象有data.adj_t 参数，给了我稀疏邻接矩阵。如何从中获得 edge_index 大小为 [2, num_edges] 的张量？

【问题讨论】：

你找到答案了吗？

标签： python graph pytorch adjacency-matrix pytorch-geometric

【解决方案1】：

正如您在docs 中看到的：

由于此功能仍处于试验阶段，某些操作，例如，图池方法，可能仍需要您输入edge_index 格式。您可以通过以下方式将adj_t 转换回(edge_index, edge_attr)：
row, col, edge_attr = adj_t.t().coo()
edge_index = torch.stack([row, col], dim=0)

【讨论】：