DGL获取adj和edge_list
发布时间
阅读量：
阅读量
DGL提供了内置函数adglgraph.adj用于获取邻接矩阵adjmatrix，然而该矩阵以sparse.tensor格式存储，无法直接用于我们所需的adj形式。
u=[0,0,0,1]
v=[1,1,2,2]
graph=dgl.DGLGraph((u,v))
print('graph',graph)
adj=graph.adjacency_matrix(transpose=True)
print('adj',adj)
>>>graph DGLGraph(num_nodes=3, num_edges=4,
ndata_schemes={}
edata_schemes={})
adj tensor(indices=tensor([[0, 0, 0, 1],
[1, 1, 2, 2]]),
values=tensor([1., 1., 1., 1.]),
size=(3, 3), nnz=4, layout=torch.sparse_coo)
AI写代码python全部评论 (0)
还没有任何评论哟~