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.