ptrblck:

As you can see self.localization returns an output of [batch_size, 10, 3, 3] . That are exactly the in_features of self.fc_loc .


I was reading that tutorial and came across your comment. I can’t understand why self.localization returns 10*3*3 features and not 10 features (like the convolution layer before it)?