DGL同构图的创建、保存、加载

创建一个简单的DGL同构图

import dgl
import torch

# 边 0->1, 0->2, 0->3, 1->3
u, v = torch.tensor([0, 0, 0, 1]), torch.tensor([1, 2, 3, 3])
g = dgl.graph((u, v))
#节点特征
g.ndata['feat'] = torch.randn(4, 64)
#节点标签
g.ndata['label'] = torch.tensor([0,1,1,0])
print(g)

运行结果

保存到本地

dgl.save_graphs("graph.pt", g)

从本地加载图数据

g = dgl.load_graphs("graph.pt")[0][0]
print(g)
print(g.ndata['feat'].shape)
print(g.ndata['label'].shape)
u, v = g.edges()
print("u=",u)
print("v=",v)