Solved by ptrblck in post #2 
              
              
                I assume the conved.squeeze() might remove unwanted dimensions during inference, since coved has only two dimensions after the squeeze. 
I’m not sure, which dimensions you would like to squeeze in this line of code, but note that all dimensions with size==1 will be removed. 
If you are using a batch…