From e6f2fc64c153247b993b8ee75e91ddf3600c7234 Mon Sep 17 00:00:00 2001 From: guofei <52460041+gfwm2013@users.noreply.github.com> Date: Sat, 5 Oct 2019 09:26:19 +0800 Subject: [PATCH] Rewrite the content of batch_cn.rst develop=test (#1372) * fix io_batch develop=test * fix io_batch develop=test * rewrite batch_cn.rst test=develop * rewrite batch_cn.rst test=develop --- doc/fluid/api_cn/io_cn/batch_cn.rst | 33 +++++++++++++++++++++++------ 1 file changed, 26 insertions(+), 7 deletions(-) diff --git a/doc/fluid/api_cn/io_cn/batch_cn.rst b/doc/fluid/api_cn/io_cn/batch_cn.rst index 914a3e754..d93e4b1d5 100644 --- a/doc/fluid/api_cn/io_cn/batch_cn.rst +++ b/doc/fluid/api_cn/io_cn/batch_cn.rst @@ -5,18 +5,37 @@ batch .. py:function:: paddle.fluid.io.batch(reader, batch_size, drop_last=False) -该层是一个batched reader。 +该接口是一个reader的装饰器。返回的reader将输入reader的数据打包成指定的batch_size大小的批处理数据(batched data)。 参数: - - **reader** (Variable)-读取数据的数据reader - - **batch_size** (int)-批尺寸 - - **drop_last** (bool) - 如果最后一个batch不等于batch_size,则drop最后一个batch。 + - **reader** (generator)- 读取数据的数据reader。 + - **batch_size** (int)- 批尺寸。 + - **drop_last** (bool) - 若设置为True,则当最后一个batch不等于batch_size时,丢弃最后一个batch;若设置为False,则不会。默认值为False。 返回:batched reader -返回类型:callable - - +返回类型:generator + +**代码示例** + +.. code-block:: python + + import paddle.fluid as fluid + + def reader(): + for i in range(10): + yield i + batch_reader = fluid.io.batch(reader, batch_size=2) + + for data in batch_reader(): + print(data) + + # 输出为: + # [0, 1] + # [2, 3] + # [4, 5] + # [6, 7] + # [8, 9] -- GitLab