Solved by tom in post #5 
              
              
                For getting tiles, manually doing this is probably most efficient: 
windows = tst_tensor.view(3, 2, 3, 2).permute(0, 2, 1, 3).reshape(1, 9, 4) 
Best regards 
Thomas