Sure. So after padding, input is now of shape (nbatch, 1, h+2, w+2).
Shifting to left by 1 is input[:, :, 2:, 1:-1].
Shifting to left by 1 and to top by 1 is input[:, :, 2:, 2:].
Disclaimer: I completely theory-crafted this in my head and haven’t actually tried. But I think it should work.