Solved by tom in post #2 
              
              
                You need to keep the Parameter of shape 4 x 1 and only do the expand in .forward. 
You need reshape instead of view, as it’ll need to instantiate the larger matrix. 
Best regards 
Thomas