torch_geometric.nn.pool.max_pool_neighbor_x

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

最大池化相邻节点特征,其中data.x中的每个特征被替换为中心节点及其邻居中具有最大值的特征值。

Return type:

Data