Как параметр groups в torch.nn.conv * влияет на процесс свертки?

Question

Я хочу свернуть многоканальный тензор с одинаковым весом одного канала. Я мог бы повторить вес по размеру канала, но подумал, что может быть другой путь.

Я думал, что параметр groups может сделать эту работу. Однако я не понимаю документацию. Вот почему я хочу спросить, как параметр groups влияет на процесс свертки?

Answer 1

Советы, так как я никогда не использовал их.

Параметр группы умножает количество ядер, которое у вас обычно есть.Поэтому, если вы установите group = 2, ожидайте в 2 раза больше ядер.

Определение conv2d в группе состояний PyTorch по умолчанию равно 1.

Если вы увеличите группу, вы получите глубинную свертку, где каждый входной канал получает конкретные ядра как таковые.

Ограничение на входящие и исходящие каналы должно делиться на номер группы.

Я думаю, что в Tensorfolow вы можете прочитать документацию SeparableConv2D, поскольку это то, что эквивалентно, когда группа> 1.