batch | 您所在的位置:网站首页 › customerpn › batch |
batch¶
paddle.
batch
(
reader,
batch_size,
drop_last=False
)
[源代码]
¶
一个 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] 使用本API的教程文档¶ 版本迁移工具 版本迁移工具 |
CopyRight 2018-2019 实验室设备网 版权所有 |