Converting PyTorch MNIST Example to TensorFlow

Created using The Code Converter.

Original Code


Resulting Code