I have a tensor named
input with dimensions 64x21x21. It is a minibatch of 64 images, each 21x21 pixels. I’d like to crop each image down to 11x11 pixels. So the output tensor I want would have dimensions 64x11x11.
I’d like to crop each image around a different “center pixel.” The center pixels are given by a 2-dimensional long tensor named
center with dimensions 64x2. For image i,
center[i] gives the row index and
center[i] gives the column index for the pixel that should be at the center in the output. We can assume that the center pixel is always at least 5 pixels away from the border.
Is there an efficient way to do this in pytorch (on the gpu)?
[NB: I posted this on stackexchange earlier, but I suspect I’m more likely to get a answer here.]