官方文档解释
groups
: 控制输入和输出之间的连接, group=1
,输出是所有的输入的卷积;group=2
,此时相当于有并排的两个卷积层,每个卷积层计算输入通道的一半,并且产生的输出是输出通道的一半,随后将这两个输出连接起来。
默认值 groups = 1
即普通卷积
groups的值必须能整除in_channels,groups的值也必须能整除out_channels
groups
: 控制输入和输出之间的连接, group=1
,输出是所有的输入的卷积;group=2
,此时相当于有并排的两个卷积层,每个卷积层计算输入通道的一半,并且产生的输出是输出通道的一半,随后将这两个输出连接起来。
即普通卷积
groups的值必须能整除in_channels,groups的值也必须能整除out_channels