ChannelShuffle

class torch.nn.ChannelShuffle(groups) [исходный код]

Разделяет и переставляет каналы тензора.

Эта операция разделяет каналы тензора формы $(N, C, *)$ на g групп в виде $(N, \frac{C}{g}, g, *)$ и перемешивает их, сохраняя исходную форму тензора в итоговом результате.

Параметры:

groups (int) – количество групп, на которые нужно разделить каналы.

Примеры:

>>> channel_shuffle = nn.ChannelShuffle(2)
>>> input = torch.arange(1, 17, dtype=torch.float32).view(1, 4, 2, 2)
>>> input
tensor([[[[ 1.,  2.],
          [ 3.,  4.]],
         [[ 5.,  6.],
          [ 7.,  8.]],
         [[ 9., 10.],
          [11., 12.]],
         [[13., 14.],
          [15., 16.]]]])
>>> output = channel_shuffle(input)
>>> output
tensor([[[[ 1.,  2.],
          [ 3.,  4.]],
         [[ 9., 10.],
          [11., 12.]],
         [[ 5.,  6.],
          [ 7.,  8.]],
         [[13., 14.],
          [15., 16.]]]])

extra_repr() [исходный код]

Возвращает дополнительное представление модуля.

Тип возвращаемого значения:

str

forward(input) [исходный код]

Выполняет прямой проход.

Тип возвращаемого значения:

Tensor