diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index 4e724838295c29195b7227dcb387ca15a4a96fa4..3899baf5e4803ef0d46d86e215348941f18703c1 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -709,8 +709,9 @@ def avg_pool2d(
         output tensor of shape :math:`(N, C, H_{\text{out}}, W_{\text{out}})`.
 
     Examples:
-        >>> input = Tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4))
-        >>> F.avg_pool2d(input, kernel_size=2, stride=2, padding=[1,0], mode="average")
+        >>> import numpy as np
+        >>> inp = Tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4))
+        >>> F.avg_pool2d(inp, kernel_size=2, stride=2, padding=[1,0], mode="average")
             Tensor([[[[0.25 1.25]
              [6.5  8.5 ]]]], device=xpux:0)
     """
diff --git a/imperative/python/megengine/module/pooling.py b/imperative/python/megengine/module/pooling.py
index f089ea95f69aa0f4175ce2618a851d194fc34721..1345e54aba7e536490a4aef7eced4477d346e37f 100644
--- a/imperative/python/megengine/module/pooling.py
+++ b/imperative/python/megengine/module/pooling.py
@@ -88,6 +88,15 @@ class AvgPool2d(_PoolNd):
         mode: whether to include the padding values while calculating the average, set
             to "average" will do counting.
             Default: "average_count_exclude_padding"
+
+    Examples:
+        >>> import numpy as np
+        >>> m = M.AvgPool2d(kernel_size=2, stride=2, padding=[1,0], mode="average")
+        >>> inp = mge.tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4))
+        >>> output = m(inp)
+        Tensor([[[[0.25 1.25]
+                  [6.5  8.5 ]]]], device=xpux:0)
+
     """
 
     def __init__(