Struct HookedDataLoader 

pub struct HookedDataLoader<T>
where
    T: Element,
{ /* private fields */ }

Sampler で選んだサンプルへ transform／collate フックをかけて バッチ化するローダー（PyTorch DataLoader(sampler=, collate_fn=) 相当。イシュー #2182）。内部に SamplerDataLoader<TensorDataset<T>> を持つ（合成）。issue の callback シグネチャ（Tensor 単体）に合わせ TensorDataset<T> 限定とする（タプルデータセットでのサンプル単位フックは対象外。 docs/tensor-core-data-sampler-hooks-decision.md §2.2・§6）。

処理順（AC-4）

HookedBatches::next は「サンプリング → サンプルごとに transform → collate」の順で処理する。transform も collate も未設定なら Dataset::batch への fast path を通る（コピー 1 回。[Self:: with_transform]／Self::with_collate を一度でも呼ぶと slow path に切り替わる）。

Implementations

impl<T> HookedDataLoader<T>

where T: Element,

pub fn new( dataset: TensorDataset<T>, sampler: impl Sampler + 'static, ) -> Result<HookedDataLoader<T>, DataError>

SamplerDataLoader::new と同じ契約（タプルではないため validate() は常に Ok(())）。

pub fn with_transform( self, f: impl Fn(Tensor<T>) -> Tensor<T> + Send + Sync + 'static, ) -> HookedDataLoader<T>

builder: infallible な transform を設定する（issue の字句どおり のシグネチャ）。内部では Ok で包んで保持する。

pub fn with_try_transform( self, f: impl Fn(Tensor<T>) -> Result<Tensor<T>, DataError> + Send + Sync + 'static, ) -> HookedDataLoader<T>

builder: fallible な transform を設定する。

pub fn with_collate( self, f: impl Fn(&[Tensor<T>]) -> Tensor<T> + Send + Sync + 'static, ) -> HookedDataLoader<T>

builder: infallible な collate を設定する（issue の字句どおりの シグネチャ）。

pub fn with_try_collate( self, f: impl Fn(&[Tensor<T>]) -> Result<Tensor<T>, DataError> + Send + Sync + 'static, ) -> HookedDataLoader<T>

builder: fallible な collate を設定する。

pub fn dataset(&self) -> &TensorDataset<T>

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

pub fn into_dataset(self) -> TensorDataset<T>

内部の dataset を取り出す。

pub fn num_batches(&self) -> Option<usize>

1 epoch あたりのバッチ数（Sampler::num_batches が Some を 返す場合のみ）。

pub fn iter(&mut self) -> HookedBatches<'_, T>

1 epoch 分のイテレータを返す。

Trait Implementations

impl<T> Debug for HookedDataLoader<T>

where T: Element,

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

Formats the value using the given formatter.