mindspore/docs/api/api_python/nn/mindspore.nn.PixelUnshuffle...

23 lines
1.3 KiB
ReStructuredText
Raw Normal View History

2022-10-22 15:19:19 +08:00
mindspore.nn.PixelUnshuffle
============================
.. py:class:: mindspore.nn.PixelUnshuffle(downscale_factor)
2023-02-06 16:55:54 +08:00
`x` 应用逆像素重组操作这是像素重组的逆操作。关于PixelUnshuffle算法详细介绍请参考 `Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network <https://arxiv.org/abs/1609.05158>`_
2022-10-22 15:19:19 +08:00
通常情况下输入shape :math:`(*, C, H \times r, W \times r)` 输出shape :math:`(*, C \times r^2, H, W)``r` 是缩小因子。 `*` 是大于等于0的维度。
参数:
2023-02-06 16:55:54 +08:00
- **downscale_factor** (int) - 恢复输入Tensor的因子是正整数。 `downscale_factor` 是上面提到的 :math:`r`
2022-10-22 15:19:19 +08:00
输入:
- **x** (Tensor) - Tensorshape为 :math:`(*, C, H \times r, W \times r)` 。输入Tensor的维度需要大于2并且倒数第一和倒数第二维length可以被 `downscale_factor` 整除。
输出:
- **output** (Tensor) - Tensorshape为 :math:`(*, C \times r^2, H, W)`
异常:
- **ValueError** - `downscale_factor` 不是正整数。
- **ValueError** - 输入 `x` 倒数第一和倒数第二维度的length不能被 `downscale_factor` 整除。
- **TypeError** - 输入 `x` 维度小于3。