dgl.distributed.in_subgraph

dgl.distributed.in_subgraph(g, nodes)[source]

返回由给定节点的入边诱导的子图。

子图保留相同的类型模式，并且所有节点都会被保留，无论它们是否拥有边。

节点/边特征不会被保留。提取的边的原始 ID 会作为返回的图中的 dgl.EID 特征存储。

目前，我们仅支持具有一个节点类型和一个边类型的输入图。

参数:

  • g (DistGraph) – 分布式图结构。

  • nodes (tensor or dict) – 用于采样邻居的节点 ID。

返回值:

子图。

可以通过子图的 dgl.EID 边特征来检索从子图边 ID 到父边 ID 的映射。

返回类型:

DGLGraph