batch

paddle.fluid.io.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.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]