B
    J4¥df  ã               @   s   g Z ddd„ZdS )Fc                s2   ‡ ‡‡fdd„}t ˆ ƒ‰ ˆ dkr.td ˆ ¡ƒ‚|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]
    c              3   sV   ˆƒ } g }x,| D ]$}|  |¡ t|ƒˆ kr|V  g }qW ˆdkrRt|ƒdkrR|V  d S )NFé    )ÚappendÚlen)ÚrÚbÚinstance)Ú
batch_sizeÚ	drop_lastÚreader© ú0/tmp/pip-unpacked-wheel-1nh_tb94/paddle/batch.pyÚbatch_reader7   s    

zbatch.<locals>.batch_readerr   zDbatch_size should be a positive integer value, but got batch_size={})ÚintÚ
ValueErrorÚformat)r	   r   r   r   r
   )r   r   r	   r   Úbatch   s    %r   N)F)Ú__all__r   r
   r
   r
   r   Ú<module>   s   