神经网络的常用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

相关推荐
说私域1 分钟前
基于开源AI智能名片链动2+1模式商城系统的创始人个人品牌资产构建研究
人工智能·小程序·开源
Ai财富密码3 分钟前
AI for Coding:如何构建基于 SDD 的多人协作流水线?
开发语言·人工智能·python
hk11248 分钟前
【Adversarial/Forensics】2026年度对抗性攻击防御与持久化内存取证基准索引 (Benchmark Index)
人工智能·网络安全·数据分析·系统架构·数据集
Rabbit_QL11 分钟前
【深度学习基础】为什么卷积层可以视为线性映射?—从结构展开到统计等价
人工智能·深度学习
颖风船16 分钟前
经典论文的ai有趣解读:Cramér-Rao下界(1945)
人工智能·信号处理·千问
Hcoco_me21 分钟前
大模型面试题34:Transformer的Encoder和Decoder区别与协作
人工智能·rnn·深度学习·lstm·transformer·word2vec
dhdjjsjs25 分钟前
Day56 PythonStudy
人工智能·深度学习·机器学习
是毛毛吧28 分钟前
AI开发工具----碾压 Bolt.new?Lovable:全栈开发的下一个“卷王”级神器深度评测
人工智能·开源·github
weixin_4379881233 分钟前
范式智能+壁仞科技|完成百余款模型适配 性能优势显著
人工智能·科技
CoovallyAIHub38 分钟前
2026计算机视觉如何将海量图像数据转化为商业价值与竞争优势?边缘计算?多模态AI?合成数据?
深度学习·算法·计算机视觉