Struct IterableDataLoader 

pub struct IterableDataLoader<D>
where
    D: IterableDataset,
{ /* private fields */ }

IterableDataset を batch_size ごとに束ねて供給するローダー （PyTorch DataLoader(iterable_ds, batch_size=k, drop_last=d) 相当）。

Implementations

impl<D> IterableDataLoader<D>

where D: IterableDataset,

pub fn new( dataset: D, batch_size: usize, drop_last: bool, ) -> Result<IterableDataLoader<D>, DataError>

batch_size == 0 は DataError::ZeroBatchSize。

pub fn dataset(&self) -> &D

保持しているデータセットへの参照。

pub fn into_dataset(self) -> D

データセットを取り出す。

pub fn batch_size(&self) -> usize

1 バッチあたりのサンプル数。

pub fn drop_last(&self) -> bool

端数バッチを捨てるか。

pub fn iter(&self) -> IterableBatches<'_, D>

1 epoch 分のイテレータを返す。呼ぶたびに IterableDataset::iter_samples で新しいストリームを開始する。