torch_geometric.nn.pool.avg_pool_neighbor_x

avg_pool_neighbor_x(data: Data, flow: Optional[str] = 'source_to_target') Data[source]

Average pools neighboring node features, where each feature in data.x is replaced by the average feature values from the central node and its neighbors.