• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1mindspore.ops.ChannelShuffle
2============================
3
4.. py:class:: mindspore.ops.ChannelShuffle(group)
5
6    将shape为 :math:`(*, C, H, W)` 的Tensor的通道划分成 :math:`g` 组，并按如下方式重新排列 :math:`(*, \frac C g, g, H*W)` ，同时保持原始Tensor的shape不变。
7
8    .. warning::
9        这是一个实验性API，后续可能修改或删除。
10
11    更多详情请查看： :class:`mindspore.ops.channel_shuffle` 。
12
13    参数：
14        - **group** (int) - 通道划分数目。
15
16    输入：
17        - **x** (Tensor) - 被划分输入Tensor。shape为 :math:`(*, C, H, W)` ，数据类型为float16, float32、int8、int16、int32、int64、uint8、uint16、uint32或uint64。
18
19    输出：
20        Tensor，数据类型与 `x` 相同，shape为 :math:`(*, C, H, W)` 。
21