nn.Conv2d()中的groups分组参数

1.参考文章:

【Pytorch】搞懂nn.Conv2d的groups参数的作用 - 知乎 (zhihu.com)

2.理解:

(1)只要你 明白了 多通道的卷积是如何实现的(可以看我的1X1卷积文章),那么这里的分组进行卷积就非常好理解了

核心:其实,虽然我们知道nn.Conv2d(in_fea,out_fea)就是从in_fea个channels 变成 out_fea个channels, 而且也知道是 由out_fea个kernel干的这件事情。

但是,每个kernel其实会把 "同一个"位置的 "所有输入channels"全部进行处理。 √

(2)效果:

--如果分groups =2 ,就可以然参数量 变成1/2 上面那篇知乎文章最后的手绘图清晰

--如果分groups =in_dim,自然,可以将参数量 变成1/in_dim

相关推荐
Black蜡笔小新3 小时前
企业AI算力工作站DLTM深度学习推理工作站零代码私有化重塑企业AI落地新模式
人工智能·深度学习
2601_959480153 小时前
Moneta Markets亿汇:“比特币反弹走势仍脆弱”
人工智能
没事别瞎琢磨3 小时前
六、输出捕获与截断
人工智能·node.js
嘉子的秃头日记3 小时前
TRO 2026|轮椅也能“猜到”用户想往哪走?
大数据·人工智能·机器学习
2601_957190904 小时前
极致裸眼沉浸!飞行影院重塑文旅游玩新体验
大数据·人工智能·旅游
Meinianda4 小时前
我用Agent 使用瑞幸官方MCP下了一单:过程全记录,优缺点分析
人工智能
没事别瞎琢磨4 小时前
七、敏感路径预检——Protected Paths
人工智能·node.js
啦啦啦_99994 小时前
4. Transformer_4_输出部分
人工智能·深度学习·transformer
用户600071819104 小时前
【翻译】构建 Claude Code 的经验:我们如何使用 Skills
人工智能
没事别瞎琢磨4 小时前
五、进程执行——spawn、超时与进程树清理
人工智能·node.js