The use case explains itself, so whenever you would need to remove a size 1 dimension, you could use squeeze. E.g. if you are dealing with a single sample somewhere in your code and would like to remove the batch size in order to plot the image tensor via plt.imshow.