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