Hi Jilee,

If you have a 1 x 152 x 138 x 32 tensor, you can take each slice of 152 x 138 x 1 and display it as a black-and white image. You can simply stack each slice three times to get 152 x 138 x 3 and feed this to pyplot’s imshow function.

PhD. Interests include Deep (Reinforcement) Learning, Computational Neuroscience, and Phenomenology.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store