def forward(self, input):
    '\n        input: batchSize * c * k*w * k*h\n        downscale_factor: k\n        batchSize * c * k*w * k*h -> batchSize * k*k*c * w * h\n        '
    return pixel_unshuffle(input, self.downscale_factor)
input: batchSize * c * k*w * k*h
downscale_factor: k
batchSize * c * k*w * k*h -> batchSize * k*k*c * w * h
def forward(self, input):
    '\n        input: batchSize * c * k*w * k*h\n        downscale_factor: k\n        batchSize * c * k*w * k*h -> batchSize * k*k*c * w * h\n        '
    return pixel_unshuffle(input, self.downscale_factor)