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