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.