我的数据对象有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(使用前将#替换为@)