Solved by albanD in post #2 
              
              
                Hi, 
The simplest solution to get these is for you only to store a 1D Tensor that contains [w0, w1…, w7] and then reconstruct the filters on every forward before giving them to F.conv2d.