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