StreamReader

class torchdata.datapipes.iter.StreamReader(datapipe, chunk=None)

给定 IO 流及其标签名称，以标签名称作为元组生成字节。

(函数名称: read_from_stream).

参数:

• datapipe – 可迭代的数据管道提供标签/URL 和字节流

• chunk – 每次迭代从流中读取的字节数。如果为 None，则将读取所有字节，直到文件末尾。

示例

>>> # xdoctest: +SKIP
>>> from torchdata.datapipes.iter import IterableWrapper, StreamReader
>>> from io import StringIO
>>> dp = IterableWrapper([("alphabet", StringIO("abcde"))])
>>> list(StreamReader(dp, chunk=1))
[('alphabet', 'a'), ('alphabet', 'b'), ('alphabet', 'c'), ('alphabet', 'd'), ('alphabet', 'e')]