torch_geometric.nn.pool.global_max_pool
- global_max_pool(x: Tensor, batch: Optional[Tensor], size: Optional[int] = None) Tensor [source]
Returns batch-wise graph-level-outputs by taking the channel-wise maximum across the node dimension.
For a single graph \(\mathcal{G}_i\), its output is computed by
\[\mathbf{r}_i = \mathrm{max}_{n=1}^{N_i} \, \mathbf{x}_n.\]Functional method of the
MaxAggregation
module.- Parameters:
x (torch.Tensor) – Node feature matrix \(\mathbf{X} \in \mathbb{R}^{(N_1 + \ldots + N_B) \times F}\).
batch (torch.Tensor, optional) – The batch vector \(\mathbf{b} \in {\{ 0, \ldots, B-1\}}^N\), which assigns each element to a specific example.
size (int, optional) – The number of examples \(B\). Automatically calculated if not given. (default:
None
)