Python PyTorch ChannelShuffle用法及代碼示例

本文簡要介紹python語言中 torch.nn.ChannelShuffle 的用法。

用法:

class torch.nn.ChannelShuffle(groups)

參數：

groups(int) -要劃分頻道的組數。

將形狀為 (*, C , H, W) 的張量中的通道劃分為 g 個組，並將它們重新排列為 (*, C \frac g, g, H, W) ，同時保持原始張量形狀。

例子：

>>> channel_shuffle = nn.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, 10],
   [11, 12]],
  [[5, 6],
   [7, 8]],
  [[13, 14],
   [15, 16]],
 ]]