diff --git a/doc/fluid/api_cn/layers_cn/gather_cn.rst b/doc/fluid/api_cn/layers_cn/gather_cn.rst index 269a56a9b2ff031730eefd4fff51f51267bd5d71..9f213831c830179963e3da868c67ff413602f40a 100644 --- a/doc/fluid/api_cn/layers_cn/gather_cn.rst +++ b/doc/fluid/api_cn/layers_cn/gather_cn.rst @@ -45,7 +45,7 @@ gather import paddle.fluid as fluid x = fluid.layers.data(name='x', shape=[-1, 5], dtype='float32') - index = fluid.layers.data(name='index', shape=[-1, 1], dtype='int32') + index = fluid.layers.data(name='index', shape=[1], dtype='int32') output = fluid.layers.gather(x, index)