mindspore.mint.nn.PixelShuffle

class mindspore.mint.nn.PixelShuffle(upscale_factor)

根据上采样系数重新排列Tensor中的元素。

更多参考详见 mindspore.mint.nn.functional.pixel_shuffle()。

支持平台：

Ascend

样例：

>>> 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]