Pytorch mnist tutorial.