Given a batch image tensor like B x C x W x H,
I would like to create a new tensor in which the new channels are the channels from nearby pixels.
For instance,
