Update mnist.py

added rng_ints