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

代码示例

  1. import paddle.fluid as fluid
  2.  
  3. def reader():
  4. for i in range(10):
  5. yield i
  6. batch_reader = fluid.io.batch(reader, batch_size=2)
  7.  
  8. for data in batch_reader():
  9. print(data)
  10.  
  11. # 输出为:
  12. # [0, 1]
  13. # [2, 3]
  14. # [4, 5]
  15. # [6, 7]
  16. # [8, 9]