Python PyTorch UnBatcher用法及代码示例

本文简要介绍python语言中 torchdata.datapipes.iter.UnBatcher 的用法。

用法:

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

参数：

• datapipe-可迭代DataPipe是un-batched

• unbatch_level-默认为1(仅展平顶层)。如果设置为 2 ，它将展平前两个级别，而 -1 将展平整个 DataPipe。

撤消数据批处理(函数名称：unbatch)。换句话说，它将批处理 DataPipe 中的数据展平至指定级别。

示例

>>> 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]