What exactly do you mean by patch?
If you have x-coordinates in the tensor x and y-coordinates in the tensor y you can do the following:
batched_images[128, 3, x, y]

and then reshape it for your requirements.