Method decorate_sample_list_generator

python/paddle/base/reader.py:1495–1566

Set the data source of the PyReader object. The provided :code:`reader` should be a Python generator, which yields list(numpy.ndarray) typed batched data. :code:`places` must be set when the PyReader object is iterable. Args: reader (generator)

(self, reader, places=None)

        )

    def decorate_sample_list_generator(self, reader, places=None):
        '''
        Set the data source of the PyReader object.

        The provided :code:`reader` should be a Python generator,
        which yields list(numpy.ndarray) typed batched data.

        :code:`places` must be set when the PyReader object is iterable.

        Args:
            reader (generator): Python generator that yields
                list(numpy.ndarray)-typed batched data.
            places (None|list(CUDAPlace)|list(CPUPlace)): place list. Must
                be provided when PyReader is iterable.

        Example:
            .. code-block:: pycon

                >>> import paddle
                >>> import paddle.base as base
                >>> import numpy as np

                >>> paddle.enable_static()

                >>> EPOCH_NUM = 3
                >>> ITER_NUM = 15
                >>> BATCH_SIZE = 3

                >>> def network(image, label):
                ...     # User-defined network, here is an example of softmax regression.
                ...     predict = paddle.static.nn.fc(x=image, size=10, activation='softmax')
                ...     return paddle.nn.functional.cross_entropy(
                ...         input=predict,
                ...         label=label,
                ...         reduction='none',
                ...         use_softmax=False,
                ...     )

                >>> def random_image_and_label_generator(height, width):
                ...     def generator():
                ...         for i in range(ITER_NUM):
                ...             fake_image = np.random.uniform(
                ...                 low=0,
                ...                 high=255,
                ...                 size=[height, width],
                ...             )
                ...             fake_label = np.ones([1])
                ...             yield fake_image, fake_label
                ...
                ...     return generator

                >>> image = paddle.static.data(name='image', shape=[None, 784, 784], dtype='float32')
                >>> label = paddle.static.data(name='label', shape=[None, 1], dtype='int64')
                >>> reader = base.io.PyReader(feed_list=[image, label], capacity=4, iterable=True)

                >>> user_defined_generator = random_image_and_label_generator(784, 784)
                >>> reader.decorate_sample_list_generator(
                ...     paddle.batch(user_defined_generator, batch_size=BATCH_SIZE),