python基础教程：对pytorch的函数中的group参数的作用介绍

@本文来源于公众号：csdn2299，喜欢可以关注公众号 程序员学府
今天小编就为大家分享一篇对pytorch的函数中的group参数的作用介绍，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
1.当设置group=1时：

conv = nn.Conv2d(in_channels=6, out_channels=6, kernel_size=1, groups=1)
conv.weight.data.size()

返回：

torch.Size([6, 6, 1, 1])

另一个例子：

conv = nn.Conv2d(in_channels=6, out_channels=3, kernel_size=1, groups=1)