神经网络汇聚层

文章目录

最大汇聚层

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

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

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),自适应池化层会自动计算池化窗口的大小和步长,以适应输入特征图的尺寸。

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

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

相关推荐
打不了嗝 ᥬ᭄几秒前
一镜通古今:Rokid AI Glasses 驱动的古建筑文物全流程智能讲解终端
人工智能
格林威7 分钟前
如何用 eBPF 监控 GigE Vision 相机网络性能
网络·人工智能·数码相机·yolo·计算机视觉·视觉检测·工业相机
云浪9 分钟前
从 0 到 1 搭建 RAG 应用:用 LangChain + Chroma + qwen-plus 实现《红楼梦》问答
javascript·vue.js·人工智能
MY_TEUCK28 分钟前
从零开始:使用Sealos Devbox快速搭建云原生开发环境
人工智能·spring boot·ai·云原生·aigc
计算机魔术师32 分钟前
【AI面试八股文 Vol.1.1 | 专题3:State Schema 设计】State Schema设计:TypedDict / Pydantic类型约束
linux·人工智能·面试
Polaris_T33 分钟前
2026最新字节大模型岗面经汇总(多平台整理)
人工智能·经验分享·算法·aigc·求职招聘
源码之家33 分钟前
计算机毕业设计:Python降水量分析与预警平台 Flask框架 数据分析 可视化 大数据 AI 大模型 爬虫 数据大屏(建议收藏)✅
人工智能·python·信息可视化·数据分析·django·flask·课程设计
游了个戏41 分钟前
用AI做了个小游戏(二)
人工智能·游戏·微信
༺ཌༀ傲世万物ༀད༻41 分钟前
如何运用好DeepSeek为自己服务:智能增强的范式革命 || 3.3 元认知强化路径
人工智能·机器学习