讲师中心 微信公众号
AI工具推荐 视频效率加速

PyTorch中Conv2d参数怎么设置_理解padding与stride对输出的影响

千辰大大_1863

千辰大大_1863

发布时间:2026-03-26 10:42:01

|

608人浏览过

|

来源于php中文网

原创

Conv2d输出尺寸由padding和stride的物理意义决定:padding是每边补零数,stride是卷积核移动步长;公式为⌊(H_in+2p−k)/s⌋+1,需实测验证。

pytorch中conv2d参数怎么设置_理解padding与stride对输出的影响

Conv2d 输出尺寸怎么算?别背公式,看懂 padding 和 stride 的物理意义

输出尺寸不是靠死记公式推出来的,而是由卷积核“怎么滑”决定的。关键就两点:stride 控制每次滑多远,padding 决定输入边缘要不要补零。补得少、滑得快,输出就小;补得多、滑得慢,输出就大。

常见错误是把 padding 当成“加几行几列”,其实它是**每边补多少**(比如 padding=1 是上下左右各补 1 行/列);stride 也不是“跳几个像素”,而是**步长单位是 kernel 移动的格子数**,默认为 1 就是挨个移。

  • 输入尺寸 H_in × W_in,卷积核大小 k,则不考虑 padding 和 stride 时,最大能放下的位置数是 H_in - k + 1
  • padding=p 后,输入等效变成 (H_in + 2p) × (W_in + 2p),再套上滑动逻辑
  • stride=s 时,实际可停的位置数是 floor((H_in + 2p - k) / s) + 1(向下取整)

padding='same' 在 PyTorch 里不存在?用对称 padding 模拟它

TensorFlow/Keras 里的 padding='same' 意思是让输出尺寸和输入一致(假设 stride=1)。PyTorch 没这个字符串选项,但可以手动算出需要的 padding 值来达到同样效果。

注意:PyTorch 的 Conv2d 只接受整数或 4 元组((top, bottom, left, right)),不支持自动对齐。如果你硬设 padding='same',会直接报错 TypeError: expected int, tuple or list。

  • 当 stride=1 且 kernel 大小为奇数(如 3、5)时,用 padding=(k-1)//2 即可实现 same 效果
  • 若 kernel 是偶数(如 4),无法做到完全对称,必须选 padding=1 或 2,输出尺寸会有 1 像素偏差
  • 想偷懒?写个辅助函数:def same_padding(kernel_size): return (kernel_size - 1) // 2,但只适用于奇数 kernel

stride > 1 时,padding 不只是“保尺寸”,还影响感受野连续性

很多人以为 stride=2 就是下采样,加不加 padding 无所谓。其实不然:不加 padding 会导致卷积核从图像左上角开始跳着覆盖,边缘信息容易丢失;加了之后,虽然输出尺寸变化不大,但第一行/列的响应更“居中”,特征对齐更好。

PyTorch Linux版 2.11.0
PyTorch Linux版 2.11.0

PyTorch 2.11.0 历史版本下载,来自 PyPI 官方发布,适合旧项目兼容、实验复现和指定环境安装。

下载

典型场景是 Encoder-Decoder 结构(如 U-Net),编码器用 stride=2 下采样,解码器用转置卷积上采样。如果编码器没加 padding,解码后会出现 1 像素偏移,拼接时 feature map 对不齐。

  • stride=2 时,推荐至少设 padding=1(尤其 kernel=3),否则输出尺寸骤减,且边界响应变弱
  • 验证方法:用全 1 输入过一层 Conv2d,打印输出,看是否四周有明显衰减——有,说明 padding 不足
  • 极端情况:stride=3, kernel=3, padding=0 → 输入 7×7,输出只有 3×3,中间区域被反复扫到,边缘只扫一次

nn.Conv2d 初始化时漏掉 bias=False,可能引发 shape 不匹配错误

这不是 padding 或 stride 的问题,但常和它们一起出现:当你手动计算输出尺寸并做后续操作(如 reshape 或拼接)时,如果忘了 bias=False,而模型里又用了 BatchNorm,就会触发 RuntimeError: Given groups=1, weight of size [o,i,k,k], expected input[1,i,h,w] to have i channels, but got X channels instead 这类迷惑错误。

原因往往是:你改了 in_channels 却没同步更新 weight 维度,而 bias 默认存在会让参数量多一维,间接干扰 shape 推导。

  • 显式写 bias=False 更安全,尤其在自定义模块或复现论文结构时
  • 检查方式:打印 layer.weight.shape 和 layer.bias.shape(如果存在),确认是否符合预期
  • 搭配 BatchNorm2d 时,bias 几乎总是冗余的,关掉还能省点显存

事情说清了就结束。最常被忽略的是:padding 和 stride 的组合效果必须在真实输入上跑一遍 print(out.shape) 来验证,光靠公式算容易在奇偶、边界、batch 维度上栽跟头。

热门AI工具

更多
二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1831

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4864

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1769

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

29017

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3467

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3487

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1223

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

616

2023.08.10

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

20

2026.10.28

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn