Pytorch mnist.