dgl.NodeFlow.block_eid

NodeFlow.block_eid(block_id)[source]

Get the edge Ids in the specified block.

The returned edge Ids are unique in the NodeFlow.

Parameters

block_id (int) – the specified block to get edge Ids.

Returns

The edge ids of the block in the NodeFlow.

Return type

Tensor