Solved by Unity05 in post #2 
              
              
                Hi, 
in your linear layer definition, you should not specify the batch size ( *in_features* , *out_features* , *bias=True* [Docs]). This does most likely cause your error. However, your concatentation does not make sense to me as your just do view(-1) effectively. You could concat compressed layer o…