神经网络汇聚层

文章目录

最大汇聚层

汇聚窗口从输入张量的左上角开始,从左往右、从上往下的在输入张量内滑动。在汇聚窗口到达的每个位置,它计算该窗口中输入子张量的最大值或平均值。计算最大值或平均值是取决于使用了最大汇聚层还是平均汇聚层。

可以设定一个任意大小的矩形汇聚窗口,并分别设定填充和步幅的高度和宽

python 复制代码
pool2d = nn.MaxPool2d((2, 3), stride=(2, 3), padding=(0, 1))

平均汇聚层

python 复制代码
 nn.AvgPool2d(kernel_size=2, stride=2)

自适应平均池化层

python 复制代码
nn.AdaptiveAvgPool2d()

这个层可以生成任意大小的输出特征图,而不需要指定池化窗口的大小。

你只需要指定输出特征图的尺寸(output_size),自适应池化层会自动计算池化窗口的大小和步长,以适应输入特征图的尺寸。

这种方式使得模型能够更好地适应不同尺寸的输入,同时保持输出尺寸的一致性。

自适应池化层特别适用于需要固定尺寸输出的场景,例如在处理不同分辨率的图像时,或者在将特征图送入全连接层之前。

相关推荐
CareyWYR9 小时前
每周AI论文速递(260323-260327)
人工智能
guoji77889 小时前
安全与对齐的深层博弈:Gemini 3.1 Pro 安全护栏与对抗测试深度拆解
人工智能·安全
实在智能RPA9 小时前
实在 Agent 和通用大模型有什么不一样?深度拆解 AI Agent 的感知、决策与执行逻辑
人工智能·ai
独隅10 小时前
PyTorch 模型部署的 Docker 配置与性能调优深入指南
人工智能·pytorch·docker
lihuayong10 小时前
OpenClaw 系统提示词
人工智能·prompt·提示词·openclaw
黑客说10 小时前
AI驱动剧情,解锁无限可能——AI游戏发展解析
人工智能·游戏
踩着两条虫10 小时前
AI驱动的Vue3应用开发平台深入探究(十):物料系统之内置组件库
android·前端·vue.js·人工智能·低代码·系统架构·rxjava
小仙女的小稀罕10 小时前
听不清重要会议录音急疯?这款常见AI工具听脑AI精准转译
开发语言·人工智能·python
reesn10 小时前
qwen3.5 0.8B纠正任务实践
人工智能·语言模型