解析平面卷积/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

其中各方法含义

  • 通道相加 就像把一本书压成一张纸,每个数都相加
  • 通道堆叠 把通道相加后的纸订成一本书,通道合并起来
相关推荐
AGI_Eval8 小时前
UniHetero:在200M+大规模数据下,生成任务能否促进视觉理解?
人工智能
CoovallyAIHub8 小时前
超越Sora的开源思路:如何用预训练组件高效训练你的视频扩散模型?(附训练代码)
深度学习·算法·计算机视觉
YuTaoShao8 小时前
【Prompt】Prompt 工程入门指南
人工智能·llm·prompt·提示词
玖日大大8 小时前
Qoder 全维度解析:AI 驱动的下一代编程生产力工具
人工智能
2301_800256118 小时前
【数据库pgsql】车辆轨迹分析视图的创建和查询代码解析
人工智能·算法·机器学习
小宇的天下8 小时前
innovus Flip chip 产品设计方法(2)
人工智能
熊猫钓鱼>_>8 小时前
AI 加 CloudBase 帮我从零快速打造儿童英语故事学习乐园
ide·人工智能·ai·mcp·codebuddy·cloudbase·ai toolkit
QYZL_AIGC8 小时前
全域众链AI赋能实体,开启数字化转型新生态
大数据·人工智能
SCKJAI8 小时前
推出高效能机器人边缘人工智能(AI)平台 ARC6N0 T5X
大数据·人工智能
新加坡内哥谈技术8 小时前
软件工程未来两年
人工智能