2022-10-20 20:05:15 +08:00
|
|
|
|
mindspore.nn.ChannelShuffle
|
|
|
|
|
|
============================
|
|
|
|
|
|
|
2022-11-01 11:21:08 +08:00
|
|
|
|
.. py:class:: mindspore.nn.ChannelShuffle(groups)
|
2022-10-20 20:05:15 +08:00
|
|
|
|
|
2023-02-27 11:22:07 +08:00
|
|
|
|
将shape为 :math:`(*, C, H, W)` 的Tensor的通道划分成 :math:`g` 组，得到shape为 :math:`(*, C \frac g, g, H, W)` 的Tensor，
|
|
|
|
|
|
并沿着 :math:`C`，:math:`\frac g` 和 :math:`g` 对应轴进行转置，将Tensor还原成原有的shape。
|
2022-10-20 20:05:15 +08:00
|
|
|
|
|
|
|
|
|
|
参数：
|
2022-11-02 09:20:24 +08:00
|
|
|
|
- **groups** (int) - 划分通道的组数。取值范围是 :math:`(0, \inf)` 。在上述公式中表示为 :math:`g` 。
|
2022-10-20 20:05:15 +08:00
|
|
|
|
|
|
|
|
|
|
输入：
|
|
|
|
|
|
- **x** (Tensor) - Tensor的shape :math:`(*, C_{in}, H_{in}, W_{in})` 。
|
|
|
|
|
|
|
|
|
|
|
|
输出：
|
|
|
|
|
|
Tensor，数据类型和shape与 `x` 相同。
|
|
|
|
|
|
|
|
|
|
|
|
异常：
|
|
|
|
|
|
- **TypeError** - `groups` 非整数。
|
|
|
|
|
|
- **ValueError** - `groups` 小于1。
|
|
|
|
|
|
- **ValueError** - `x` 的维度小于3。
|
|
|
|
|
|
- **ValueError** - Tensor的通道数不能被 `groups` 整除。
|
