Thanks for the help!!! I really appreciate it.
This works perfectly - When reconstructing the patches, I just want to recover the original image without the padding so I’m just slicing re to get what I need.
Can you recommend a good tutorial or blog post to get more familiar with these tensor manipulation methods in PyTorch? The docs aren’t the most helpful at times.