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

其中各方法含义

  • 通道相加 就像把一本书压成一张纸,每个数都相加
  • 通道堆叠 把通道相加后的纸订成一本书,通道合并起来
相关推荐
哥布林学者1 分钟前
吴恩达深度学习课程五:自然语言处理 第三周:序列模型与注意力机制(一)seq2seq 模型
深度学习·ai
数字游民952712 分钟前
小程序上新,猜对了么更新110组素材
人工智能·ai·小程序·ai绘画·自媒体·数字游民9527
泰迪智能科技25 分钟前
分享|联合编写教材入选第二批“十四五”职业教育国家规划教材名单
大数据·人工智能
Yongqiang Cheng27 分钟前
PyTorch Grid Sample
pytorch·grid sample
模型时代36 分钟前
热力学计算技术或将大幅降低AI图像生成能耗
人工智能
企业老板ai培训37 分钟前
从九尾狐AI实战案例拆解AI短视频获客的架构设计:智能矩阵如何提升企业效率?
人工智能
龙腾AI白云1 小时前
知识图谱如何在制造业实际落地应用
人工智能·知识图谱
力学与人工智能1 小时前
“高雷诺数湍流数据库的构建及湍流机器学习集成研究”湍流重大研究计划集成项目顺利结题
数据库·人工智能·机器学习·高雷诺数·湍流·重大研究计划·项目结题
娟宝宝萌萌哒1 小时前
智能体设计模式重点
人工智能·设计模式
gsgbgxp1 小时前
WSL迁移至非系统盘
深度学习·ubuntu