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

其中各方法含义

  • 通道相加 就像把一本书压成一张纸,每个数都相加
  • 通道堆叠 把通道相加后的纸订成一本书,通道合并起来
相关推荐
tellmewhoisi7 分钟前
机器学习:集成学习4(XGBoost推导)
人工智能·机器学习·集成学习
鲜于言悠90511 分钟前
AI自动化工作流实战:告别重复加班的四层架构方案
人工智能
一水鉴天12 分钟前
软件方法—工程函数—语言模型 20260925(千问)
人工智能·机器学习·边缘计算
程序员的账号13 分钟前
《Python工匠》资源分享
人工智能·python·深度学习·机器学习
揽秀亭长15 分钟前
视频转脚本有哪些技术路线?三种常见方案对比分析
人工智能·音视频
2601_9622029816 分钟前
首衡集采集配怕破损?万象包装优化降损耗
大数据·人工智能
随性而行36018 分钟前
企业微信二次开发如何接入大模型工具?API接口实现智能任务调用的技术思路
java·前端·人工智能·python·微信·机器人·企业微信
IT大白鼠19 分钟前
彭大帅的AI运维助手——自然语言管理 Linux 集群与网络设备——第 0 篇 · 导读:把 Linux 运维交给 AI,到底靠谱吗
linux·运维·人工智能
程序员的账号20 分钟前
《深度学习入门2自制框架》中文PDF+源代码+斋藤康毅
人工智能·深度学习·pdf
一木 之林23 分钟前
多模态大模型一统精讲 NLP 和 CV:从 ViT、CLIP 到 BLIP、LLaVA 的三段式统一架构
人工智能·自然语言处理·架构