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]