Graph Readout

The problem, from first principles

Graph Readout aggregates node-level representations into a single graph-level representation. Common readout functions include sum, mean, and max pooling.

Where READOUT is a permutation-invariant aggregation function.

From theory to code

Implement:

graph_readout(node_features, readout_type='sum')

Aggregates node features into a graph-level representation.

Constraints

• node_features shape: (num_nodes, feature_dim).
• readout_type in ('sum', 'mean', 'max').
• Return shape: (feature_dim,).