batch(reader, batch_size, drop_last=False)
This operator creates a batched reader which combines the data from the input reader to batched data.
reader (generator) – the data reader to read from.
batch_size (int) – size of each mini-batch.
drop_last (bool, optional) – If set to True, the last batch is dropped when the size of last batch is not equal to batch_size, if set to False, it will not. Default: False.
The batched reader.
- Return Type:
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) # Output is # [0, 1] # [2, 3] # [4, 5] # [6, 7] # [8, 9]