Your activation (act) seems to have a wrong size and plt.imshow cannot visualize it.
Make sure you are trying to plot a tensor in the shape [height, width, 3] or [height, width] and pass it as a numpy array to plt.imshow via tensor.numpy().