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

相关推荐
尹玉玺5 分钟前
事实判断与价值判断的融合辨析:厘清道法自然与该当本然的边界
人工智能
9i编程6 分钟前
用 Trae 吃透掘金文章:从拉取 Demo 到跑起 Three.js 太阳能电场,AI 问出来的学习路径
人工智能·openai·ai编程
廋到被风吹走14 分钟前
DeepSeek V4.1-Flash-1017:一次“既要又要还要“的升级
人工智能
lll235269177915 分钟前
采访内容如何从录音变成可用资料?
人工智能
weixin_4387193915 分钟前
AI 改配置:语法全对,语义全错——三道闸门照着抄
人工智能
haluhalu.19 分钟前
深入理解AIAgent(1.3)--- 设计模式
人工智能·python·设计模式·langchain
stereohomology22 分钟前
外挂可能是用AI工具自定义solomd插件和主题适合的方式
人工智能·markdown·薅羊毛·solomd·外挂新特性
东离与糖宝33 分钟前
一个句号击穿Copilot安全机制,CVSS8.8高危漏洞深度解析
人工智能
“AI国潮设计-小江”41 分钟前
《【AIGC实战】Python+SDXL批量生成“英歌舞树莓香草布丁”IP,附保持角色一致性的代码逻辑》
开发语言·人工智能·python·prompt·aigc