Add record shuffling to Problem.dataset and batch shuffling.
Add record shuffling to Problem.dataset and batch shuffling.

PiperOrigin-RevId: 218558666

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor