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

相关推荐
It's now3 小时前
Spring AI 基础开发流程
java·人工智能·后端·spring
Glad_R4 小时前
巧用AI流程图,让信息呈现更全面
人工智能·信息可视化·产品运营·流程图·产品经理
西南胶带の池上桜4 小时前
1.Pytorch模型应用(线性与非线性预测)
人工智能·pytorch·python
杀生丸学AI4 小时前
【无标题】VGGT4D:用于4D场景重建的视觉Transformer运动线索挖掘
人工智能·深度学习·3d·aigc·transformer·三维重建·视觉大模型
小和尚同志4 小时前
还在手动配置?这款开源软件让你一键配置 Claude Code 和 Codex
人工智能·aigc
阿正的梦工坊5 小时前
ProRL:延长强化学习训练,扩展大语言模型推理边界——NeurIPS 2025论文解读
人工智能·语言模型·自然语言处理
致Great5 小时前
Ollama 进阶指南
人工智能·gpt·chatgpt·agent·智能体
Nautiluss5 小时前
一起玩XVF3800麦克风阵列(八)
大数据·人工智能·嵌入式硬件·github·音频·语音识别
yzx9910135 小时前
人工智能大模型新浪潮:五大突破性工具深度解析
人工智能