Python PyTorch Header用法及代码示例

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

用法:

class torchdata.datapipes.iter.Header(source_datapipe: IterDataPipe[torchdata.datapipes.iter.util.header.T_co], limit: int = 10)

参数：

• source_datapipe-将产生元素的DataPipe

• limit-停止前要产生的元素数

从源 DataPipe 开始生成元素，直到指定的限制(函数名称：header )。

示例

>>> from torchdata.datapipes.iter import IterableWrapper
>>> dp = IterableWrapper(range(10))
>>> header_dp = dp.header(3)
>>> list(header_dp)
[0, 1, 2]