神经网络汇聚层

文章目录

最大汇聚层

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

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

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

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

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

相关推荐
undsky_4 分钟前
【n8n教程】:执行工作流——从手动测试到生产自动化
人工智能·ai·aigc·ai编程
牛客企业服务5 分钟前
牛客AI面试蓝领案例:破解制造业招聘效率困局
人工智能·面试·职场和发展
oscar9996 分钟前
深度学习测试题与解析
人工智能·深度学习·测试题
Wis4e6 分钟前
基于PyTorch的深度学习——迁移学习4
pytorch·深度学习·迁移学习
小oo呆6 分钟前
【自然语言处理与大模型】LangChainV1.0入门指南:核心组件Models
人工智能·自然语言处理
Ama_tor7 分钟前
AI-agent|从人工智能代理角度再延伸至扣子编程
人工智能
北冥有一鲲8 分钟前
LangChain 框架前世今生:从“万能接口”到“AI 应用全家桶”
人工智能·chatgpt·langchain
MuseDAM_cc8 分钟前
企业素材找不到?DAM 3 步解决资产分散
大数据·人工智能
AI即插即用8 分钟前
即插即用系列 | 2025 RestorMixer:融合 CNN、Mamba 与 Transformer 的高效图像复原的集大成者!
人工智能·深度学习·神经网络·目标检测·计算机视觉·cnn·transformer
hay_lee10 分钟前
LLaDA2.0:蚂蚁开源业内首个 100B 扩散语言模型
人工智能·语言模型·自然语言处理·llada2.0