mindspore.ops.pixel_shuffle

mindspore.ops.pixel_shuffle(input, upscale_factor)[源代码]

对输入 input 应用像素重组操作,它实现了步长为 \(1/r\) 的子像素卷积。关于pixel_shuffle算法详细介绍,请参考 Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network

通常情况下,input shape \((*, C \times r^2, H, W)\) ,输出shape \((*, C, H \times r, W \times r)\)r 是缩小因子。 * 是大于等于0的维度。

参数:
  • input (Tensor) - Tensor,shape为 \((*, C \times r^2, H, W)\)input 的维度需要大于2,并且倒数第三维length可以被 upscale_factor 的平方整除。

  • upscale_factor (int) - 打乱输入Tensor的因子,是正整数。 upscale_factor 是上面提到的 \(r\)

返回:
  • output (Tensor) - Tensor,shape为 \((*, C, H \times r, W \times r)\)

异常:
  • ValueError - upscale_factor 不是正整数。

  • ValueError - input 倒数第三维度的length不能被 upscale_factor 的平方整除。

  • ValueError - input 维度小于3。

  • TypeError - input 不是Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import ops
>>> input_x = np.arange(3 * 2 * 9 * 4 * 4).reshape((3, 2, 9, 4, 4))
>>> input_x = mindspore.Tensor(input_x, mindspore.dtype.int32)
>>> output = ops.pixel_shuffle(input_x, 3)
>>> print(output.shape)
(3, 2, 1, 12, 12)