view() unfortunately isn’t complex enough for this task…
Also: is it possible to for view() to have a similar functionality to np.reshape with the argument order ?  I know we can use permute, but it would be more convenient  e.g. Equivalent of np.reshape() in pyTorch?