Solved by ptrblck in post #2 
              
              
                I think the easiest way would be to define the parameters manually and use the functional API for all operations, e.g.: 
out = F.conv2d(intput, weight)

where you’ve defined the weight as a parameter before.