Hello,
I think it literally means what it says.
Based on your input Tensor’s size, I would venture that you might double check that you are using pytorch’s dimension convention of Batch x Channels x Height x Width. Personally, I usually find the pytorch documentation helpful, it gives details about the dimensions in most layers, e.g. for Conv2d 361.
Best regards
Thomas