.. _cn_api_fluid_layers_batch: batch ------------------------------- .. py:function:: paddle.fluid.layers.batch(reader, batch_size) 该层是一个reader装饰器。接受一个reader变量并添加``batching``装饰。读取装饰的reader,输出数据自动组织成batch的形式。 参数: - **reader** (Variable)-装饰有“batching”的reader变量 - **batch_size** (int)-批尺寸 返回:装饰有``batching``的reader变量 返回类型:变量(Variable) **代码示例**: .. code-block:: python import paddle.fluid as fluid raw_reader = fluid.layers.io.open_files(filenames=['./data1.recordio', './data2.recordio'], shapes=[(3,224,224), (1,)], lod_levels=[0, 0], dtypes=['float32', 'int64'], thread_num=2, buffer_size=2) batch_reader = fluid.layers.batch(reader=raw_reader, batch_size=5) # 如果用raw_reader读取数据: # data = fluid.layers.read_file(raw_reader) # 只能得到数据实例。 # # 但如果用batch_reader读取数据: # data = fluid.layers.read_file(batch_reader) # 每5个相邻的实例自动连接成一个batch。因此get('data')得到的是一个batch数据而不是一个实例。