mindspore.nn.ChannelShuffle

.. py:class:: mindspore.nn.ChannelShuffle(groups)

    将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。

    参数：
        - **groups** (int) - 划分通道的组数，必须大于0。在上述公式中表示为 :math:`g` 。

    输入：
        - **x** (Tensor) - Tensor的shape :math:`(*, C_{in}, H_{in}, W_{in})` 。

    输出：
        Tensor，数据类型和shape与 `x` 相同。

    异常：
        - **TypeError** - `groups` 非正整数。
        - **ValueError** - `groups` 小于1。
        - **ValueError** - `x` 的维度小于3。
        - **ValueError** - `x` 的通道数不能被 `groups` 整除。