Load mnist dataset pytorch.