graphstorm.gsf.get_node_feat_size

graphstorm.gsf.get_node_feat_size(g, node_feat_names)

Get the overall feature size of each node type with feature names specified in the node_feat_names. If a node type has multiple features, the returned feature size will be the sum of the sizes of these features for that node type.

Parameters

gDistGraph

A DGL distributed graph.

node_feat_namesstr, or dict of list of str

The node feature names. A string indicates that all nodes share the same feature name, while a dictionary with a list of strings indicates that each node type has different node feature names.

Returns

node_feat_size: dict of int

The feature size for the node types and feature names specified in the node_feat_names.