Struct DataLoader 

pub struct DataLoader<D>
where
    D: Dataset,
{ /* private fields */ }

dataset を DataLoaderConfig に従ってミニバッチへ分割する （PyTorch torch.utils.data.DataLoader 相当）。

Implementations

impl<D> DataLoader<D>

where D: Dataset,

pub fn new( dataset: D, config: DataLoaderConfig, ) -> Result<DataLoader<D>, DataError>

config.batch_size == 0 は DataError::ZeroBatchSize。 dataset.validate() が失敗した場合はその結果をそのまま返す （タプルデータセットの長さ不一致を構築時に fail-closed に拒否 する。モジュール冒頭 A08 注記）。

pub fn dataset(&self) -> &D

保持するデータセットへの参照。

pub fn config(&self) -> DataLoaderConfig

構成のコピー（DataLoaderConfig: Copy）。

pub fn len(&self) -> usize

1 epoch あたりのバッチ数。

pub fn is_empty(&self) -> bool

len() == 0 か。

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

1 epoch 分のイテレータを返す。shuffle=true のときは呼び出し ごとに順列を引き直す（PyTorch の epoch ごと再シャッフルと同義。 モジュール冒頭「シャッフル契約」参照）。

n = self.dataset.len() から確定する添字順列 Vec<usize> は checked_numel_for::<usize>(&[n]) で確保前にバイトサイズまで 検査してから構築する。Dataset::len() は実装者が任意の値を 返せるため（例: 要素数ゼロの Tensor::new(vec![], &[usize::MAX, 0]) から len() == usize::MAX のデータセットを作れる）、検査 なしに (0..n).collect()／shuffled_indices の Vec::with_ capacity を呼ぶと numel * size_of::<usize>() > isize::MAX で capacity overflow パニックしうる（本番経路 panic 禁止規約 .claude/rules/coding-rust.md。PR #1867 codex-review P1 是正）。

この iter() 自体は（設計どおり）Result を返さない infallible な API のまま維持する。検査に失敗した場合は order を空のまま にし、shuffle=true でも with_global_rng を呼ばず（RNG を一切 消費しない）、DataError::Shape（ShapeError:: ElementCountOverflow）を Batches へ持ち越して最初の next() 呼び出しで 1 回だけ Err として yield し、以降は None を返す（ExactSizeIterator::len() もこの 1 件に整合させ る。下記 Batches::next 参照）。

pub fn into_dataset(self) -> D

内部の dataset を取り出す。

Trait Implementations

impl<D> Debug for DataLoader<D>

where D: Dataset,

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl<'a, D> IntoIterator for &'a DataLoader<D>

where D: Dataset,

type Item = Result<<D as Dataset>::Batch, DataError>

The type of the elements being iterated over.

type IntoIter = Batches<'a, D>

Which kind of iterator are we turning this into?

fn into_iter(self) -> <&'a DataLoader<D> as IntoIterator>::IntoIter

Creates an iterator from a value.