batch

paddle.fluid.io.batch(reader, batch_size, drop_last=False)[源代码]

该接口是一个reader的装饰器。返回的reader将输入reader的数据打包成指定的batch_size大小的批处理数据（batched data）。

参数

• reader (generator)- 读取数据的数据reader。
• batch_size (int)- 批尺寸。
• drop_last (bool) - 若设置为True，则当最后一个batch不等于batch_size时，丢弃最后一个batch；若设置为False，则不会。默认值为False。

返回

batched reader

返回类型

generator

代码示例

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]