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