I hope you don’t mind me following up on this question with a related one! Tensor unfolding + your example is definitely what I’m looking for, but I was wondering if you had any ideas on ways to visualize/verify the blocks so I can double check that the blocks include the correct sections that I want (especially to double check if I’m using fold in the proper order of dimensions), as well as how they join together to make the whole block.