通道洗牌操作

假设我们有一个4通道的2x2的图像(batch size为1)，即b=1, c=4, h=w=2，并且我们选择groups=2，那么我们将通道分为两组。

给定以下输入x：

x = 
[
 [[1, 2],  [3, 4]],
 [[5, 6],  [7, 8]],
 [[9, 10], [11, 12]],
 [[13, 14],[15, 16]]
]

首先，我们把x重塑成5个维度，其形状是(1, 2, 2, 2, 2)，也就是说，我们有两个组，每组有两个通道。

x = 
[
  [
    [[1, 2],  [3, 4]],
    [[5, 6],  [7, 8]]
  ],
  [
    [[9, 10], [11, 12]],
    [[13, 14],[15, 16]]
  ]
]

接下来，我们使用permute来交换groups和channels/groups的维度：

x = 
[
  [
    [[1, 2],  [3, 4]],
    [[9, 10], [11, 12]]
  ],
  [
    [[5, 6],  [7, 8]],
    [[13, 14],[15, 16]]
  ]
]

最后，我们将x重新整形成其原始的(1, 4, 2, 2)形状：

x = 
[
 [[1, 2],  [3, 4]],
 [[9, 10], [11, 12]],
 [[5, 6],  [7, 8]],
 [[13, 14],[15, 16]]
]

你可以看到，现在的通道顺序是[1, 3, 2, 4]，原来的相邻通道1和2现在被分散了。这就完成了通道的洗牌。