diff --git a/python/paddle/nn/functional/input.py b/python/paddle/nn/functional/input.py
index 2e4bbd99a726d7c85fb18054e75b1c3cbcfaf394..f6f367bf23d0826551c35904a21aff392a481f24 100644
--- a/python/paddle/nn/functional/input.py
+++ b/python/paddle/nn/functional/input.py
@@ -74,15 +74,14 @@ def one_hot(x, num_classes, name=None):
 
             import paddle
             # Correspond to the first example above, where label.shape is 4 and one_hot_label.shape is [4, 4].
-            label = paddle.static.data(name="label", shape=[4, 1], dtype="int64")
+            label = paddle.to_tensor([1, 1, 3, 0], dtype='int64')
             # label.shape = [4]
-            # label.data = [1, 1, 3, 0]
-            one_hot_label = paddle.nn.functional.one_hot(x=label, num_classes=4)
+            one_hot_label = paddle.nn.functional.one_hot(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.]]
+            # one_hot_label = [[0., 1., 0., 0.],
+            #                  [0., 1., 0., 0.],
+            #                  [0., 0., 0., 1.],
+            #                  [1., 0., 0., 0.]]
 
     """