Solved by Azerus in post #2 
              
              
                Hi, 
Yes, you can use view() safely here, or reshape. 
See this for the differences between view and reshapeView is quick to perform but will fails on non-contiguous tensors. 
Reshape always works. When possible, the returned tensor will be a view of input. Otherwise, it will be a copy. 
But I don’t…