batch

paddle. batch ( reader, batch_size, drop_last=False ) [source]

This operator creates a batched reader which combines the data from the input reader to batched data.

Parameters
  • 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.

Returns

The batched reader.

Return Type:

generator

Examples

>>> 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]