diff --git a/python/paddle/nn/initializer/assign.py b/python/paddle/nn/initializer/assign.py
index ac6dc3f0f31c7a682806d1cddf41646f43d1b612..a33301230e89e14b4d0d7c87bf7fa2dcc55ef179 100644
--- a/python/paddle/nn/initializer/assign.py
+++ b/python/paddle/nn/initializer/assign.py
@@ -51,7 +51,7 @@ class Assign(NumpyArrayInitializer):
             # linear_1.weight:  [2. 2.]
             # linear_1.bias:  [2.]
 
-            res_1 = linear(data_1)
+            res_1 = linear_1(data_1)
             # res_1:  [6.]
 
             # python list
@@ -66,7 +66,7 @@ class Assign(NumpyArrayInitializer):
             # linear_2.weight:  [2. 2.]
             # linear_2.bias:  [2.]
 
-            res_2 = linear(data_2)
+            res_2 = linear_2(data_2)
             # res_2:  [6.]
 
             # tensor
@@ -81,7 +81,7 @@ class Assign(NumpyArrayInitializer):
             # linear_3.weight:  [2. 2.]
             # linear_3.bias:  [2.]
 
-            res_3 = linear(data_3)
+            res_3 = linear_3(data_3)
             # res_3:  [6.]
     """