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