AvgPooling¶

class dgl.nn.mxnet.glob.AvgPooling[source]

Bases: mxnet.gluon.block.Block

Apply average pooling over the nodes in the graph.

$r^{(i)} = \frac{1}{N_i}\sum_{k=1}^{N_i} x^{(i)}_k$
forward(graph, feat)[source]

Compute average 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