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)