def __init__(self, downscale_factor):
    super(PixelUnShuffle, self).__init__()
    self.downscale_factor = downscale_factor
def __init__(self, downscale_factor):
    super(PixelUnShuffle, self).__init__()
    self.downscale_factor = downscale_factor