dgl.NodeFlow.node_attr_schemes¶
-
NodeFlow.
node_attr_schemes
(layer_id)[source]¶ Return the node feature schemes.
Each feature scheme is a named tuple that stores the shape and data type of the node feature
- Parameters
layer_id (int) – the specified layer to get node data scheme.
- Returns
The schemes of node feature columns.
- Return type
dict of str to schemes