ChannelShuffle(2) >>> input = torch.randn(1, 4, 2, 2) >>> print(input) [[[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 10], [11, 12]], [[13, 14], [15, 16]], ]] >>> output = channel_shuffle(input) >>> print(output) [[[1, 2], [3, 4]], [[9, ...