torch_geometric.nn.pool.max_pool_neighbor_x

max_pool_neighbor_x(data: Data, flow: Optional[str] = 'source_to_target') → Data

Max pools neighboring node features, where each feature in data.x is replaced by the feature value with the maximum value from the central node and its neighbors.