Refactor MNIST forward style and improve device selection#1419

amit-chaubey wants to merge 1 commit intopytorch:mainpytorch/examples:mainfrom

amit-chaubey:cleanup-mnist-forward-device-selectionamit-chaubey/pytorch-examples:cleanup-mnist-forward-device-selectionCopy head branch name to clipboard