megengine.functional.nn.pixel_shuffle# pixel_shuffle(inp, upscale_factor)[source]# Rearranges elements in a tensor of shape (…, C * r^2, H, W) to a tensor of shape (…, C, H * r, W * r), where r is an upscale factor, where … is zero or more batch dimensions. Parameters: inp (Tensor) – input tensor. upscale_factor (int) – upscale factor of pixel_shuffle. Return type: Tensor Returns: output tensor.