注:借鉴整理,仅供自学,侵删
pytorch::Dataloader中的迭代器和生成器应用详解
pytorch中的数据加载模块 Dataloader,使用生成器来返回数据的索引,使用迭代器来返回需要的张量数据,可以在大量数据情况下,实现小批量循环迭代式的读取,避免了内存不足问题。
注:借鉴整理,仅供自学,侵删
pytorch::Dataloader中的迭代器和生成器应用详解
pytorch中的数据加载模块 Dataloader,使用生成器来返回数据的索引,使用迭代器来返回需要的张量数据,可以在大量数据情况下,实现小批量循环迭代式的读取,避免了内存不足问题。