Python PyTorch Concater用法及代碼示例

本文簡要介紹python語言中 torchdata.datapipes.iter.Concater 的用法。

用法:

class torchdata.datapipes.iter.Concater(*datapipes: IterDataPipe)

參數：

datapipes-可迭代 DataPipes 被連接

連接多個 Iterable DataPipes (函數名稱：concat )。生成的 DataPipe 將生成第一個輸入 DataPipe 中的所有元素，然後再生成後續的元素。

示例

>>> import random
>>> from torchdata.datapipes.iter import IterableWrapper
>>> dp1 = IterableWrapper(range(3))
>>> dp2 = IterableWrapper(range(5))
>>> list(dp1.concat(dp2))
[0, 1, 2, 0, 1, 2, 3, 4]