diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index f1b466e0a9d9e2797b29a327623eed14a1108721..d2acbc1bf179143b2384581e8f5eeb4a0ac88ff2 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -668,7 +668,7 @@ def max_pool2d(
 
     Examples:
         >>> import numpy as np
-        >>> input = tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4))
+        >>> input = Tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4))
         >>> F.nn.max_pool2d(input, 2, 1, 0)
         Tensor([[[[ 5.  6.  7.]
            [ 9. 10. 11.]]]], device=xpux:0)