pub fn generate_batch_indices( num_samples: usize, batch_size: usize, shuffle: bool, ) -> Vec<usize>
Generate random batch indices