torch_geometric.nn.pool.max_pool_x

max_pool_x(cluster: Tensor, x: Tensor, batch: Tensor, batch_size: Optional[int] = None, size: Optional[int] = None) → Tuple[Tensor, Optional[Tensor]][source]

根据cluster中定义的聚类，对节点特征进行最大池化。

Parameters:

cluster (torch.Tensor) – 聚类向量 \(\mathbf{c} \in \{ 0, \ldots, N - 1 \}^N\)，它将每个节点分配到一个特定的聚类。
x (Tensor) – 节点特征矩阵。
batch (torch.Tensor) – 批次向量 \(\mathbf{b} \in {\{ 0, \ldots, B-1\}}^N\)，它将每个节点分配给一个特定的示例。
batch_size (int, optional) – The number of examples \(B\). Automatically calculated if not given. (default: None)
size (int, optional) – 单个示例中的最大聚类数。此属性对于获取批处理密集表示非常有用，例如用于应用FC层，但应仅在已知每个示例的最大聚类数的情况下使用。（默认值：None）

Return type: