o Me@sgZdddZdS)Fcs2fdd}tdkrtd|S)a This operator creates a batched reader which combines the data from the input reader to batched data. Args: 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: .. code-block:: python 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) # Output is # [0, 1] # [2, 3] # [4, 5] # [6, 7] # [8, 9] c3s\}g}|D]}||t|kr|Vg}qdur*t|dkr,|VdSdSdS)NF)appendlen)rbinstance batch_size drop_lastreader.batch_readerrzFbatch_size should be a positive integeral value, but got batch_size={})int ValueErrorformat)r rr r r rr batchs% rN)F)__all__rr r r r s