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

2023-12-14

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


正如你在docs:

由于此功能仍处于实验阶段,一些操作,e.g.,图池方法,可能仍然需要您输入edge_index格式。你可以转换adj_t回到(edge_index, edge_attr) via:

row, col, edge_attr = adj_t.t().coo()
edge_index = torch.stack([row, col], dim=0)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

Pytorch几何稀疏邻接矩阵到边缘索引张量 的相关文章

随机推荐