dgl-learn

#默认建图是64位,但是可以用32位来加快速度
g=dgl.graph((u,v),num_nodes=8, idtype=th.int32)
bg=dgl.to_bidirected(g)
g.nodes()
g.edges()

#特征信息
g.ndata['x']=th.ones(g.num_nodes(),3)
g.edata['x']=th.ones(g.num_nodes(),dtype=th.int32)
g.ndata['y']
g.ndata['x'][1]
g.ndata['x'][th.tensor([2,3])]
g.edata['w']=weights

#加载图
dgl.from_scipy()

#gpu使用DGL graph
cuda_g=g.to(device)
#如果u,v均为gpu tensor,那么建的图也在gpu上
#如果图在gpu上,那么接受的数据类型也必须在gpu上,比如索引
复制