squeeze will remove all unit axis which can lead to bugs (for instance, you might not be able to predict on a single example).
tensor.squeeze() is there any advantage to explicitly state which dimension you want to squeeze?
Answer
squeeze will remove all unit axis which can lead to bugs (for instance, you might not be able to predict on a single example).
Relevant part of lecture