当前位置: 首页 > news >正文

PyTorch神经网络组件之Conv2d

nn.Conv2d

Conv2d 本质一句话:用多个卷积核扫描图像,提取特征。

Conv2d 输入必须是 4维张量:batch_size, in_channels, height, width。

例如:
x shape = (1, 3, 32, 32)
表示:
1张图片 3通道(RGB) 32x32大小

Conv2d:

nn.Conv2d(in_channels, out_channels, kernel_size)
例如:
conv = nn.Conv2d(3, 16, 3)
表示:输入通道 = 3输出通道 = 16卷积核大小 = 3×3
输出:
(1, 16, 30, 30)输出大小计算公式:
output_height =(input_height - kernel_size) + 1

Conv2d 权重结构(核心)

Conv2d 内部权重 shape:out_channels, in_channels, kernel_height, kernel_width

例如: conv = nn.Conv2d(3,16,3)
weight shape: (16, 3, 3, 3)
表示:16个卷积核每个卷积核:3通道每通道 3×3

卷积核结构可视化

kernel 1:channel 1:[ w11 w12 w13 ][ w14 w15 w16 ][ w17 w18 w19 ]channel 2:[ w21 ... ]channel 3:[ w31 ... ]

所以一个卷积核shape: (3, 3, 3)

卷积计算过程理解

输入对象数据:
1通道 4×4:[1 2 3 4][5 6 7 8][9 1 2 3][4 5 6 7]
卷积核: 3x3[1 0 0][0 1 0][0 0 1]计算左上角:
1×1 + 2×0 + 3×0
5×0 + 6×1 + 7×0
9×0 + 1×0 + 2×1= 1+6+2 = 9
得到输出像素。

总结

Conv2d 本质:

输入:
(batch, in_channels, H, W)权重:
(out_channels, in_channels, kernel, kernel)输出:
(batch, out_channels, H-k+1, W-k+1)
http://www.jsqmd.com/news/424933/

相关文章:

  • 中考阅读理解丢分严重?3家靠谱线上机构实测推荐,帮娃高效提分不踩坑 - 品牌测评鉴赏家
  • 看2026上海家装,原木风装修厂家推荐排行来啦,轻奢大平层设计/新中式软装设计/原木风精装房设计,上海家装厂家有哪些 - 品牌推荐师
  • 2026初中语文阅读理解线上辅导大揭秘,选对机构成绩飙升! - 品牌测评鉴赏家
  • 这次终于选对AI论文工具,千笔·专业学术智能体 VS 云笔AI,专科生专属!
  • Task04:字符串
  • 261_尚硅谷_单元测试综合案例
  • 基于YALMIP和CPLEX的Matlab程序实现配电网故障重构模型的二阶锥优化方法,通用性高...
  • 用while循环输出1~1000之间能被5整除的数,并且每行输出3个
  • 实测才敢推!MBA专属降AI神器 —— 千笔·专业降AI率智能体
  • 初中语文提分秘籍!揭秘3家极速提分线上机构 - 品牌测评鉴赏家
  • 干货来了:千笔·降AI率助手,遥遥领先的降AI率工具
  • 2026年灰铁厂家推荐,精密圆棒生产与质量检测认证 - 品牌鉴赏师
  • 小学语文阅读理解辅导哪个平台好?实测5大热门平台,家长闭眼冲不踩坑 - 品牌测评鉴赏家
  • Copilot 模式和 Agent 模式的区别是什么?
  • 论文写不动?千笔ai写作,人气爆表的AI论文网站
  • std::stringstream 超详细讲解(C++ 字符串流)
  • 【ISAC】6G低空经济新突破!协作式ISAC技术实现无人机精准感知与参数估计【附MATLAB代码】
  • 什么是向量数据库?在基于大模型的应用开发中,向量数据库主要解决什么问题?
  • 用for循环输出1~1000之间能被5整除的数,并且每行输出3个
  • 你都了解哪些向量数据库?如何选型?
  • 家长必看!2026宝藏线上机构,助力孩子作文稳步提升 - 品牌测评鉴赏家
  • 家人们谁懂啊!孩子作文像流水账,这些线上机构救大命 - 品牌测评鉴赏家
  • 计算0~100的奇数和、偶数和、总和
  • 向量数据库原理是什么? 请简述下它的原理
  • 45232453
  • 2026年球墨铸铁厂家推荐,资质齐全与售后完善优质企业 - 品牌鉴赏师
  • 计算机毕业设计springboot河北水利电力学院班级事务管理系统 基于SpringBoot的高校班级综合事务管理平台 基于Java的院系班级日常管理与学生服务系统
  • 小学生写作不再愁!3家宝藏线上机构大放送 - 品牌测评鉴赏家
  • 计算机毕业设计springboot河环院校友管理系统 基于SpringBoot的高校校友信息管理与服务平台 基于Java的毕业生联络与资源共享系统
  • 吐血整理!小学语文线上课程红黑榜来袭 - 品牌测评鉴赏家