Question 10/10 v3 lecture 8

When doing tensor.squeeze() is there any advantage to explicitly state which dimension you want to squeeze?


Yes, otherwise 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