You could initialize the input tensor with:
x = torch.arange(batch_size*C*H*W)..view(batch_size, C, H, W).float()

so that you can visually inspect each patch and make sure it contains the right values.
I’m not sure I understand the last sentence. Would you like to fold them back to the original input afterwards?