diff --git a/doc/fluid/api_cn/layers_cn/dynamic_gru_cn.rst b/doc/fluid/api_cn/layers_cn/dynamic_gru_cn.rst
index 60c47755901e5460b4b5d41f7c43075a2fd686cd..081e84c40a996e36d11d4e5690b202dce3da58c1 100644
--- a/doc/fluid/api_cn/layers_cn/dynamic_gru_cn.rst
+++ b/doc/fluid/api_cn/layers_cn/dynamic_gru_cn.rst
@@ -60,9 +60,11 @@ dynamic_gru
     import paddle.fluid as fluid
 
     dict_dim, emb_dim = 128, 64
-    data = fluid.layers.data(name='sequence', shape=[1],
-                             dtype='int32', lod_level=1)
-    emb = fluid.layers.embedding(input=data, size=[dict_dim, emb_dim])
+    data = fluid.data(name='sequence',
+                shape=[None],
+                dtype='int64',
+                lod_level=1)
+    emb = fluid.embedding(input=data, size=[dict_dim, emb_dim])
     hidden_dim = 512
     x = fluid.layers.fc(input=emb, size=hidden_dim * 3)
     hidden = fluid.layers.dynamic_gru(input=x, size=hidden_dim)
diff --git a/doc/fluid/api_cn/layers_cn/gru_unit_cn.rst b/doc/fluid/api_cn/layers_cn/gru_unit_cn.rst
index f78b317c33c6d5680ef465a9ae734a88d6d0486a..a0385a997b11402690e4758e9f88f8656afec4a3 100644
--- a/doc/fluid/api_cn/layers_cn/gru_unit_cn.rst
+++ b/doc/fluid/api_cn/layers_cn/gru_unit_cn.rst
@@ -57,12 +57,12 @@ Gated Recurrent Unit（GRU）循环神经网络计算单元。该OP用于完成
     import paddle.fluid as fluid
 
     dict_dim, emb_dim = 128, 64
-    data = fluid.layers.data(name='step_data', shape=[1], dtype='int32')
-    emb = fluid.layers.embedding(input=data, size=[dict_dim, emb_dim])
+    data = fluid.data(name='step_data', shape=[None], dtype='int64')
+    emb = fluid.embedding(input=data, size=[dict_dim, emb_dim])
     hidden_dim = 512
     x = fluid.layers.fc(input=emb, size=hidden_dim * 3)
-    pre_hidden = fluid.layers.data(
-        name='pre_hidden', shape=[hidden_dim], dtype='float32')
+    pre_hidden = fluid.data(
+        name='pre_hidden', shape=[None, hidden_dim], dtype='float32')
     hidden = fluid.layers.gru_unit(
         input=x, hidden=pre_hidden, size=hidden_dim * 3)