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

相关推荐
后端小张9 分钟前
【AI 解析】Gemini 3 全面解析:从认知到落地
人工智能·机器学习·ai·语言模型·chatgpt·imagen·gemini
HyperAI超神经1 小时前
【TVM 教程】优化大语言模型
人工智能·语言模型·自然语言处理·cpu·gpu·编程语言·tvm
musk12121 小时前
文本分析与挖掘,nlp,中文产品评论情感分析最佳实践方案
人工智能·自然语言处理
专注数据的痴汉1 小时前
「数据获取」《中国商务年鉴》(2004-2024)
大数据·人工智能·信息可视化
limenga1021 小时前
奇异值分解(SVD):深度理解神经网络的内在结构
人工智能·深度学习·神经网络·机器学习
双向331 小时前
为什么Doubao-Seed-Code成为我的主力编程助手?实测与深度解析
人工智能
秋邱1 小时前
【机器学习】深入解析线性回归模型
人工智能·机器学习·线性回归
数字游名Tomda1 小时前
腾讯开源最新视频生成模型:仅8.3B,元宝已接入
人工智能·开源·开源软件
一点一木1 小时前
国内首款原生视觉编程模型实测:Doubao-Seed-Code 前端 Agent 从零完成像素画编辑器
前端·人工智能·agent
●VON2 小时前
人工智能、机器学习与深度学习:从概念到实践
人工智能·深度学习·机器学习