dgl.distributed.find_edges

dgl.distributed.find_edges(g, edge_ids)[source]

给定一个边ID数组，从分布式图中返回源节点和目标节点的ID数组 s 和 d。 s[i] 和 d[i] 分别是边 eid[i] 的源节点和目标节点的ID。

Parameters:

  • g (DistGraph) – The distributed graph.

  • edges (tensor) – 边ID数组。

Returns:

  • tensor – 源节点ID数组。

  • tensor – 目标节点ID数组。