mindspore.ops.channel_shuffle

mindspore.ops.channel_shuffle(x, groups)

Divide the channels in a tensor of shape $(\ast ,C,H,W)$ into $g$ groups and rearrange them as $(\ast ,\frac{C}{g},g,H\ast W)$, while retaining the original tensor shape in the final output.

Parameters

• x (Tensor) – The input tensor.

• groups (int) – Number of groups to divide channels in.

Returns

Tensor

Supported Platforms:

Ascend CPU

Examples

>>> 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]]
   [[ 8  9]
   [10 11]]
   [[ 4  5]
   [ 6  7]]
   [[12 13]
   [14 15]]]]