Batcher

class torchdata.datapipes.map.Batcher(datapipe: MapDataPipe[T], batch_size: int, drop_last: bool = False, wrapper_class=List)

创建数据的小批量（函数名称：batch）。

如果 drop_last 设置为 True，将添加一个外部维度作为 batch_size，或者如果 drop_last 设置为 False，则对最后一个批次使用 length % batch_size。

参数：:

• datapipe – 正在进行批处理的可迭代 DataPipe

• batch_size – 每个批次的大小

• drop_last – 选项，如果最后一个批次不满，则丢弃它

示例

>>> # xdoctest: +SKIP
>>> from torchdata.datapipes.map import SequenceWrapper
>>> dp = SequenceWrapper(range(10))
>>> batch_dp = dp.batch(batch_size=2)
>>> list(batch_dp)
[[0, 1], [2, 3], [4, 5], [6, 7], [8, 9]]