dgl.NodeFlow.block_edges

NodeFlow.block_edges(block_id, remap_local=False)[source]

Return the edges in a block.

If remap_local is True, returned indices u, v, eid will be remapped to local Ids (i.e. starting from 0) in the block or in the layer. Otherwise, u, v, eid are unique in the NodeFlow.

Parameters:
  • block_id (int) – The specified block to return the edges.
  • remap_local (boolean) – Remap layer/block-level local Id if True; otherwise, NodeFlow-level Id.
Returns:

  • Tensor – The src nodes.
  • Tensor – The dst nodes.
  • Tensor – The edge ids.