解析平面卷积/pytorch的nn.Conv2d的计算步骤,in_channels与out_channels如何计算而来

关于二维图像卷积详细过程/pytorch-nn.Conv2d方法

python 复制代码
nn.Conv2d(
  in_channels=inChannel, 
  out_channels=outChannel, 
  kernel_size=kernelSize, 
  stride=stride, 
  padding=padding,
)

具体详细步骤如图

详细说明

  1. 定义outChannel个卷积块,每个卷积块有inChannel个卷积核,所以一共 inChannel * outChannel个卷积核
  2. 输入块的每个通道与卷积核卷积,由此一来,每个卷积块得到一个中间块,共outChannel个块
  3. 每个中间块通道相加,每个中间块压缩成一个平面,共outChannel个平面
  4. 平面叠加起来,成一个输出块,所以输出块的通道为outChannel

其中各方法含义

  • 通道相加 就像把一本书压成一张纸,每个数都相加
  • 通道堆叠 把通道相加后的纸订成一本书,通道合并起来
相关推荐
百胜软件@百胜软件6 小时前
重塑零售未来:百胜智能中台+胜券AI,赋能品牌零售撬动3100亿增量市场
大数据·人工智能·零售
Shawn_Shawn12 小时前
人工智能入门概念介绍
人工智能
极限实验室12 小时前
程序员爆哭!我们让 COCO AI 接管 GitLab 审查后,团队直接起飞:连 CTO 都说“这玩意儿比人靠谱多了
人工智能·gitlab
Maynor99613 小时前
Z-Image: 100% Free AI Image Generator
人工智能
爬点儿啥13 小时前
[Ai Agent] 10 MCP基础:快速编写你自己的MCP服务器(Server)
人工智能·ai·langchain·agent·transport·mcp
张人玉13 小时前
百度 AI 图像识别 WinForms 应用代码分析笔记
人工智能·笔记·百度
测试人社区-小明14 小时前
智能弹性伸缩算法在测试环境中的实践与验证
人工智能·测试工具·算法·机器学习·金融·机器人·量子计算
Spring AI学习14 小时前
Spring AI深度解析(9/50):可观测性与监控体系实战
java·人工智能·spring
罗西的思考15 小时前
【Agent】MemOS 源码笔记---(5)---记忆分类
人工智能·深度学习·算法
dajun18112345615 小时前
反 AI 生成技术兴起:如何识别与过滤海量的 AI 伪造内容?
人工智能