dgl.distributed.in_subgraph

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

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

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

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

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

Parameters:

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

  • nodes (tensor 或 dict) – 从中采样邻居的节点ID。

Returns:

子图。

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

Return type:

DGLGraph