diff --git a/python/paddle/nn/functional/input.py b/python/paddle/nn/functional/input.py
index e51da1bd7445602c52db081a0f30f3549643d5bb..e77bf0e39672984f7076938b134f3e54f4c761ab 100644
--- a/python/paddle/nn/functional/input.py
+++ b/python/paddle/nn/functional/input.py
@@ -72,17 +72,17 @@ def one_hot(x, num_classes, name=None):
     Examples:
         .. code-block:: python
 
-            import paddle.fluid as fluid
+            import paddle
             # Correspond to the first example above, where label.shape is 4 and one_hot_label.shape is [4, 4].
-            label = fluid.data(name="label", shape=[4, 1], dtype="int64")
+            label = paddle.data(name="label", shape=[4, 1], dtype="int64")
             # label.shape = [4]
             # label.data = [1, 1, 3, 0]
-            one_hot_label = fluid.one_hot(x=label, num_classes=4)
+            one_hot_label = paddle.nn.functional.one_hot(x=label, num_classes=4)
             # one_hot_label.shape = [4, 4]
             # one_hot_label.data = [[0., 1., 0., 0.],
-                                    [0., 1., 0., 0.],
-                                    [0., 0., 0., 1.],
-                                    [1., 0., 0., 0.]]
+            #                       [0., 1., 0., 0.],
+            #                       [0., 0., 0., 1.],
+            #                       [1., 0., 0., 0.]]
     """
 
     if in_dygraph_mode():