I’m using the MNIST dataset for image classification as done here  https://medium.com/dair-ai/building-rnns-is-fun-with-pytorch-and-google-colab-3903ea9a3a79 (in the “RNN for image classification” bit).
The data has shape 28x28, which means that i have 28 timesteps (rows in the image) and 28 inputs (number of columns) . The data goes into the rnn row by row.
I would like my code to print the outputs after each timestep i.e. I would like the code to give me the probabilities after each row goes into the rnn.
Can this be done?