Struct WeightedRandomSampler 

pub struct WeightedRandomSampler { /* private fields */ }

重み付き復元・非復元抽出でバッチ添字を供給する（PyTorch WeightedRandomSampler 相当。イシュー #2182）。構築時 （Self::new）に crate::rng::validate_multinomial で検証する （RNG を消費しない）。start_epoch は公開 API crate::rng::multinomial（1 回の with_global_rng を使う原子的な 抽選）を呼び、同一シードの下で rng::multinomial(&weights, num_samples, replacement) と抽選列が bit 完全一致する。

Implementations

impl WeightedRandomSampler

pub fn new( weights: Vec<f32>, num_samples: usize, replacement: bool, batch_size: usize, drop_last: bool, ) -> Result<WeightedRandomSampler, DataError>

weights（rank 1: 各要素が対応する添字の重み）・抽選する サンプル総数 num_samples・復元抽出の有無・バッチ化条件を 受け取る。構築時に有限性・非負・総和が正・非復元抽出時の 重み不足を検証する（crate::rng::validate_multinomial を 再利用。RNG は一切消費しない）。batch_size == 0 は DataError::ZeroBatchSize。

Trait Implementations

impl Debug for WeightedRandomSampler

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

Formats the value using the given formatter.

impl Sampler for WeightedRandomSampler

fn start_epoch(&mut self) -> Result<(), DataError>

epoch 開始時に 1 回呼ぶ。抽選を伴う実装はここで確定させる。

fn next_batch(&mut self) -> Vec<usize>

次バッチの添字列。空の Vec は epoch 終了の番兵。

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

1 epoch あたりのバッチ数（既知なら Some。既定は None）。 SamplerBatches::size_hint・HookedBatches::size_hint は 本メソッドの戻り値（epoch 全体の静的総数）と、各イテレータ自身が 追跡する「これまでに yield した数」の差分で残数を計算する （レビュー指摘: イシュー #2182 review・PR #2310 codex-review P2 是正）。カスタム Sampler 実装が進捗追跡用の別メソッドを override する必要はなく、本メソッドだけを実装すれば途中経過・ epoch 終了後のどちらでも Iterator::size_hint の「残り要素数」 契約（下限の非過大申告）を満たす。