ptrblck:

PS: don’t forget to add a batch dimension even to a single sample.


Help understand this solution. How can I add a batch size if I have one sample?
Do I need to add the batch size with which I train the model or set the size to 1?