torch.channel_shuffle

torch.channel_shuffle(input, groups) → Tensor

Разделяет каналы тензора формы $(*, C , H, W)$ на g групп и переставляет их в форму $(*, C \frac g, g, H, W)$, сохраняя исходную форму тензора.

Подробности см. в ChannelShuffle.

Параметры:

• input (тензор) – входной тензор
• groups (int) – количество групп, на которые разделяются и переставляются каналы.

Примеры:

>>> 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 = torch.nn.functional.channel_shuffle(input, 2)
>>> print(output)
[[[[1, 2],
   [3, 4]],
  [[9, 10],
   [11, 12]],
  [[5, 6],
   [7, 8]],
  [[13, 14],
   [15, 16]],
 ]]