torch_geometric.nn.pool.avg_pool_neighbor_x

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

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

Return type:

Data