Thanks for your reply and help so far, I really appreciate it! To clarify my last sentence, is there a way to check in what order you combine each of the smaller blocks to generate the larger block? For example, when I split up my 32x32x32 grid into 8 smaller 16x16x16 grids, the returned tensor from the fold operations will just have essentially a list of the blocks, but I don’t know if those blocks were taken by going horizontally across the larger grid, or by going vertically. I could match these up by manually inspecting the values in each of the smaller blocks, and trying to check where they appear in the larger block to see, but that’s a bit tedious. Hopefully I explained that better?
The context here is that I’m trying to match up a coordinates from a separate smaller grid with each block.