将一个图划分为多个子图,使用dgl.node_subgraph()
分割的子图会多出 edges_schemes:{“_ID” …}
举例:
import dgl
import torch
src=[0,1,2,3,4,5]
dst=[1,2,3,0,5,4]
feats=torch.tensor([[0,0,0],[1,1,1],[2,2,2],[3,3,3],[4,4,4],[5,5,5]])
g=dgl.graph((torch.tensor(src),torch.tensor(dst)))
g.