mindspore.ops.channel_shuffle
- mindspore.ops.channel_shuffle(x, groups)[源代码]
将shape为 \((*, vpn永久免费梯子 C, H, W)\) 的tensor的通道划分成 \(g\) vpn free 组,并按如下方式重新排列 \((*, \frac{C}{g}, g, H*W)\) ,同时在最终输出中保持原始tensor的shape。
警告
从2.9.0(不含)之后版本开始,该接口将从函数形态迁移为 Module 免费的vpn梯子 形态,请优先使用
mindspore.ops.ChannelShuffle。- 参数:
x (Tensor) - 输入tensor。
groups (int) - 通道划分数目。
- 返回:
Tensor
- 支持平台:
AscendCPU
样例:
>>> import mindspore >>> x = mindspore.tensor(mindspore.ops.arange(0, 16, dtype=mindspore.int16).reshape(1, 4, 2, 2)) >>> y = mindspore.ops.channel_shuffle(x, groups=2) >>> print(y) [[[[ 0 1] [ 2 3]] 免费的vpn梯子 [[ 8 9] [10 11]] [[ 4 5] [ vpn free vpn梯子 免费 6 7]] [[12 13] [14 15]]]]