UnBatcher

class torchdata.datapipes.iter.UnBatcher(datapipe: IterDataPipe, unbatch_level: int = 1)

撤销数据的批处理（函数名称：unbatch）。

换句话说，它会扁平化批处理数据管道中的数据，直到指定的级别。

参数：:

• datapipe – 正在进行取消批处理的可迭代数据管道

• unbatch_level – 默认值为 1（只扁平化顶层）。如果设置为 2，它将扁平化前两层，而 -1 将扁平化整个数据管道。

示例

>>> # xdoctest: +SKIP
>>> from torchdata.datapipes.iter import IterableWrapper
>>> source_dp = IterableWrapper([[[0, 1], [2]], [[3, 4], [5]], [[6]]])
>>> dp1 = source_dp.unbatch()
>>> list(dp1)
[[0, 1], [2], [3, 4], [5], [6]]
>>> dp2 = source_dp.unbatch(unbatch_level=2)
>>> list(dp2)
[0, 1, 2, 3, 4, 5, 6]