random batch indexes, test set separately