Thanks!
I have tried it, but it requires the input images in the same batch must have the same shape.
I want to keep the images’ original shape without scale operation, so the images in a batch have different shapes, and it seems no corresponding solution.