dgl.NodeFlow.layer_nid

NodeFlow.layer_nid(layer_id)[source]

Get the node Ids in the specified layer.

The returned node Ids are unique in the NodeFlow.

Parameters:layer_id (int) – The layer to get the node Ids.
Returns:The node ids.
Return type:Tensor