forked from mindspore-Ecosystem/mindspore
25 lines
1.3 KiB
ReStructuredText
25 lines
1.3 KiB
ReStructuredText
|
mindspore.nn.PixelUnshuffle
|
|||
|
|
============================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.nn.PixelUnshuffle(downscale_factor)
|
|||
|
|
|
|||
|
|
PixelUnshuffle函数。
|
|||
|
|
|
|||
|
|
在多个输入平面组成的输入上面应用PixelUnshuffle算法。关于PixelUnshuffle算法详细介绍，请参考 `Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network <https://arxiv.org/abs/1609.05158>`_ 。
|
|||
|
|
|
|||
|
|
通常情况下，输入shape :math:`(*, C, H \times r, W \times r)` ，输出shape :math:`(*, C \times r^2, H, W)` 。`r` 是缩小因子。 `*` 是大于等于0的维度。
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
- **downscale_factor** (int) - 减小空间分辨率的因子，是正整数。
|
|||
|
|
|
|||
|
|
输入：
|
|||
|
|
- **x** (Tensor) - Tensor，shape为 :math:`(*, C, H \times r, W \times r)` 。输入Tensor的维度需要大于2，并且倒数第一和倒数第二维length可以被 `downscale_factor` 整除。
|
|||
|
|
|
|||
|
|
输出：
|
|||
|
|
- **output** (Tensor) - Tensor，shape为 :math:`(*, C \times r^2, H, W)` 。
|
|||
|
|
|
|||
|
|
异常：
|
|||
|
|
- **ValueError** - `downscale_factor` 不是正整数。
|
|||
|
|
- **ValueError** - 输入 `x` 倒数第一和倒数第二维度的length不能被 `downscale_factor` 整除。
|
|||
|
|
- **TypeError** - 输入 `x` 维度小于3。
|