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]