假设我们有一个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现在被分散了。这就完成了通道的洗牌。

08-28 15:26