MaxPooling¶
-
class
dgl.nn.mxnet.glob.MaxPooling[source]¶ Bases:
mxnet.gluon.block.BlockApply max pooling over the nodes in the graph.
\[r^{(i)} = \max_{k=1}^{N_i} \left( x^{(i)}_k \right)\]-
forward(graph, feat)[source]¶ Compute max pooling.
- Parameters
graph (DGLGraph) – The graph.
feat (mxnet.NDArray) – The input feature with shape \((N, *)\) where \(N\) is the number of nodes in the graph.
- Returns
The output feature with shape \((B, *)\), where \(B\) refers to the batch size.
- Return type
mxnet.NDArray
-