src/transformers/models/swin2sr/modeling_swin2sr.py/UpsampleOneStep/forward
class UpsampleOneStep:    def forward(self, x):
        x = self.conv(x)
        x = self.pixel_shuffle(x)

        return x