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

其中各方法含义

  • 通道相加 就像把一本书压成一张纸,每个数都相加
  • 通道堆叠 把通道相加后的纸订成一本书,通道合并起来
相关推荐
自不量力的A同学6 分钟前
Solon AI v3.9 正式发布:全能 Skill 爆发
java·网络·人工智能
一枕眠秋雨>o<12 分钟前
从抽象到具象:TBE如何重构AI算子的编译哲学
人工智能
xiaobaibai15312 分钟前
决策引擎深度拆解:AdAgent 用 CoT+RL 实现营销自主化决策
大数据·人工智能
island131414 分钟前
CANN ops-nn 算子库深度解析:神经网络核心计算的硬件映射、Tiling 策略与算子融合机制
人工智能·深度学习·神经网络
冬奇Lab15 分钟前
一天一个开源项目(第14篇):CC Workflow Studio - 可视化AI工作流编辑器,让AI自动化更简单
人工智能·开源·编辑器
是小蟹呀^16 分钟前
从稀疏到自适应:人脸识别中稀疏表示的核心演进
人工智能·分类
云边有个稻草人17 分钟前
CANN ops-nn:筑牢AIGC的神经网络算子算力底座
人工智能·神经网络·aigc·cann
island131417 分钟前
CANN Catlass 算子模板库深度解析:高性能 GEMM 架构、模板元编程与融合算子的显存管理策略
人工智能·神经网络·架构·智能路由器
结局无敌18 分钟前
从算子到生态:cann/ops-nn 如何编织一张高性能AI的协作之网
人工智能
心疼你的一切18 分钟前
数字智人:CANN加速的实时数字人生成与交互
数据仓库·深度学习·aigc·交互·cann