diff --git a/doc/fluid/api_cn/layers_cn/sampled_softmax_with_cross_entropy_cn.rst b/doc/fluid/api_cn/layers_cn/sampled_softmax_with_cross_entropy_cn.rst
index 3097aaaab7eafeaa38ceaa49d718363a7fe50e4a..2cbbed3a594dad6de440f576bae1c59b23c11ccf 100644
--- a/doc/fluid/api_cn/layers_cn/sampled_softmax_with_cross_entropy_cn.rst
+++ b/doc/fluid/api_cn/layers_cn/sampled_softmax_with_cross_entropy_cn.rst
@@ -37,7 +37,7 @@ sampled_softmax_with_cross_entropy
     import paddle.fluid as fluid
 
     input = fluid.layers.data(name='data', shape=[256], dtype='float32')
-    label = fluid.layers.data(name='label', shape=[5], dtype='int64')
+    label = fluid.layers.data(name='label', shape=[1], dtype='int64')
     fc = fluid.layers.fc(input=input, size=100)
     out = fluid.layers.sampled_softmax_with_cross_entropy(
               logits=fc, label=label, num_samples=25)