神经网络的常用layer

卷积层

nn.Conv1d

池化层

下采样层主要是使用pooling层的操作技术来实现的,目的主要是保存有效的信息,降低特征的维度,来避免过拟合。对一个M*N的图像,进行s倍的下采样,对长和宽来说各要缩小s倍

全连接层

nn.Linear

(70 封私信 / 85 条消息) 全连接层的作用是什么? - 知乎 (zhihu.com)

复制代码
torch.nn.Linear(in_features, # 输入的神经元个数
           out_features, # 输出神经元个数
           bias=True # 是否包含偏置
           )
nn.Linear(2, 1) # 输入特征数为2,输出特征数为1

Pytorch nn.Linear的基本用法与原理详解_iioSnail的博客-CSDN博客

激活层

nn.relu()

本质是max(0, input),对输入数据进行逐元素操作,如果输入的元素值小于0,则输出为0;否则,输出等于输入。

nn.BatchNormid

相关推荐
luoganttcc4 分钟前
除了视觉伺服 还有哪些 方法
人工智能
ST小智4 分钟前
2025年创作历程回顾与个人生活平衡
大数据·linux·人工智能
weixin_4379881210 分钟前
范式智能发布“风控哨兵”大模型 引领金融风控新范式
人工智能
哥本哈士奇11 分钟前
使用Gradio构建AI前端 - RAG的QA模块
前端·人工智能·状态模式
5G全域通14 分钟前
面向5G复杂性的下一代运维技术体系:架构、工具与实践
大数据·运维·人工智能·5g·架构
你们补药再卷啦16 分钟前
人工智能算法概览
人工智能·算法
悟纤19 分钟前
续写卡在 2 秒?解决方案全解析|Suno 进阶指南|第 13 篇
人工智能·suno·suno ai·suno api·ai music
RockHopper202527 分钟前
企业运营认知机器人的落地规范说明 —— 一种以工程化实现/商业化落地为目的设计原则
人工智能·llm·认知机器人·认知导向
吃人陈乐游刘28 分钟前
05实战经验X-anylabelingAI自动标注数据集-onnx简单解绍(2025年12月)
人工智能·深度学习
AndrewHZ32 分钟前
【图像处理基石】什么是光栅化?
图像处理·人工智能·算法·计算机视觉·3d·图形渲染·光栅化