fix bug that failure to add Resize to transforms

[yaojin17]

The original code does not support adding Resize function to transforms. For example, when I add transforms.Resize((32,32)) to transforms, it will report the bug RuntimeError: The expanded size of the tensor (64) must match the existing size (32) at non-singleton dimension 2. Target sizes: [3, 64, 64]. Tensor sizes: [3, 32, 32].
To make it work, I first decide if there is a Resize function and then add the if-else condition to determine the size of the last 2 dimensions of the feature_output.