PixelShuffle

class PixelShuffle(upscale_factor, **kwargs)

Rearranges elements in a tensor of shape (, C x r^2, H, W) to a tensor of shape (, C, H x r, W x r), where r is an upscale factor, where * is zero or more batch dimensions.