batch¶
该接口是一个reader的装饰器。返回的reader将输入reader的数据打包成指定的batch_size大小的批处理数据(batched data)。
警告
不推荐使用这个API,如有数据加载需求推荐使用支持多进程并发加速的 paddle.io.DataLoader
参数¶
reader (generator)- 读取数据的数据reader。
batch_size (int)- 批尺寸。
drop_last (bool) - 若设置为True,则当最后一个batch不等于batch_size时,丢弃最后一个batch;若设置为False,则不会。默认值为False。
返回¶
batched reader
代码示例¶
import paddle
def reader():
for i in range(10):
yield i
batch_reader = paddle.batch(reader, batch_size=2)
for data in batch_reader():
print(data)
# 输出为:
# [0, 1]
# [2, 3]
# [4, 5]
# [6, 7]
# [8, 9]