Concatenate dimensions in a tensor

Could you check this?

It’s a not very good but very graphical explanation about view and permute which may help you to order your data as you expect