Skip to content

Instantly share code, notes, and snippets.

@Steboss89

Steboss89/generate_random_batch.rs

Created August 14, 2022 17:27
Show Gist options
  • Save Steboss89/51e36aa4829c05df818da70b2758e47e to your computer and use it in GitHub Desktop.
Save Steboss89/51e36aa4829c05df818da70b2758e47e to your computer and use it in GitHub Desktop.
Download ZIP
Generate random batches
Raw
generate_random_batch.rs
pub fn generate_random_index(ArraySize: i64, BatchSize: i64)-> Tensor{
let random_idxs = Tensor::randint(ArraySize, &[BatchSize], kind::INT64_CPU);
random_idxs
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment