當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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]

相關用法


注：本文由純淨天空篩選整理自pytorch.org大神的英文原創作品 torchdata.datapipes.iter.Header。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。