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