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]