标题

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

从源 DataPipe 的开头生成元素，直到指定的限制（函数名称：header）。

如果您想手动将 DataPipe 的长度设置为某个值，建议您使用 LengthSetter。

参数：:

• source_datapipe – 将生成元素的 DataPipe

• limit – 在停止之前要生成的元素数量

示例

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