util.
buffer_batch
Buffer an iterable of batches into larger (or smaller) batches
generator : iterable
The generator to buffer
buffer_size : int > 0
The number of examples to retain per batch.
batch
A batch of size at most buffer_size