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

Question

我的数据对象有data.adj_t参数，给了我稀疏的邻接矩阵。我怎样才能从中得到edge_index大小的张量[2, num_edges]？

Answer 1

正如您在文档中看到的：

由于此功能仍处于试验阶段，某些操作，例如图池方法，可能仍需要您输入edge_index格式。您可以通过以下方式转换adj_t回(edge_index, edge_attr)：

row, col, edge_attr = adj_t.t().coo()
edge_index = torch.stack([row, col], dim=0)