21 lines
833 B
ReStructuredText
21 lines
833 B
ReStructuredText
mindspore.ops.ChannelShuffle
|
||
============================
|
||
|
||
.. py:class:: mindspore.ops.ChannelShuffle(group)
|
||
|
||
将shape为 :math:`(*, C, H, W)` 的Tensor的通道划分成 :math:`g` 组，并按如下方式重新排列 :math:`(*, \frac{C}{g}, g, H*W)` ，同时在最终输出中保持原始Tensor的shape。
|
||
|
||
.. warning::
|
||
这是一个实验性API，后续可能修改或删除。
|
||
|
||
更多详情请查看： :class:`mindspore.ops.channel_shuffle` 。
|
||
|
||
参数：
|
||
- **group** (int) - 通道划分数目。
|
||
|
||
输入：
|
||
- **x** (Tensor) - 被划分的输入Tensor。shape为 :math:`(*, C, H, W)` ，数据类型为float16、float32、int8、int16、int32、int64、uint8、uint16、uint32或uint64。
|
||
|
||
输出：
|
||
Tensor，数据类型与 `x` 相同，shape为 :math:`(*, C, H, W)` 。
|