神经网络汇聚层

文章目录

最大汇聚层

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

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

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

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

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

相关推荐
摆烂咸鱼~4 分钟前
机器学习(13-2)
人工智能·机器学习
人工智能AI技术4 分钟前
从零复现马斯克开源X推荐算法
人工智能
Mixtral14 分钟前
4款录音转文字工具深度评测:钉钉闪记、Otter、随身鹿、讯飞听见...AI后处理能力谁更强?
人工智能
智驱力人工智能18 分钟前
守护矿山动脉 矿山皮带跑偏AI识别系统的工程化实践与价值 皮带偏离检测 皮带状态异常检测 多模态皮带偏离监测系统
大数据·人工智能·opencv·算法·安全·yolo·边缘计算
大模型真好玩19 分钟前
大模型训练全流程实战指南基础篇(二)——大模型文件结构解读与原理解析
人工智能·pytorch·langchain
周博洋K20 分钟前
Deepseek的新论文Engram
人工智能
e***985722 分钟前
2024技术趋势:AI领跑,云端边缘共舞
人工智能
智驱力人工智能27 分钟前
构筑安全红线 发电站旋转设备停机合规监测的视觉分析技术与应用 旋转设备停机检测 旋转设备异常检测 设备停机AI行为建模
人工智能·opencv·算法·安全·目标检测·计算机视觉·边缘计算
独自破碎E29 分钟前
怎么优化RAG的检索效果?
人工智能·自然语言处理
这儿有一堆花29 分钟前
从机械傀儡到具身智能:机器人控制模型的演变实录
人工智能·机器人