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