mindspore.mint.nn.PixelShuffle

class mindspore.mint.nn.PixelShuffle(upscale_factor)

Rearrange elements in a tensor according to an upscaling factor.

For details, please refer to mindspore.mint.nn.functional.pixel_shuffle().

Supported Platforms:

Ascend

Examples

>>> from mindspore import mint
>>> pixel_shuffle = mint.nn.PixelShuffle(3)
>>> input = mint.randn(1, 9, 4, 4)
>>> output = pixel_shuffle(input)
>>> print(output.shape())
    [1, 1, 12, 12]