Thanks a lot for the clarification! I think I understand it now much better 
I was wondering if you any suggestion for this case if now I want to use fold to make it back.
Hmmm I think we also dont have x.fold in pytorch?