dgl.distributed.find_edges

dgl.distributed.find_edges(g, edge_ids)

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

参数:

• g (DistGraph) – 分布式图。

• edges (tensor) – 边 ID 数组。

返回值:

• tensor – 源节点 ID 数组。

• tensor – 目标节点 ID 数组。