dgl.NodeFlow.edge_attr_schemes

NodeFlow.edge_attr_schemes(block_id)[source]

Return the edge feature schemes.

Each feature scheme is a named tuple that stores the shape and data type of the node feature

Parameters:block_id (int) – the specified block to get edge data scheme.
Returns:The schemes of edge feature columns.
Return type:dict of str to schemes