dgl.distributed.find_edges๏ƒ

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

Given an edge ID array, return the source and destination node ID array s and d from a distributed graph. s[i] and d[i] are source and destination node ID for edge eid[i].

Parameters:

  • g (DistGraph) โ€“ The distributed graph.

  • edges (tensor) โ€“ The edge ID array.

Returns:

  • tensor โ€“ The source node ID array.

  • tensor โ€“ The destination node ID array.