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

相关推荐
wangqiaowq4 分钟前
CV 后端打包
人工智能
AKAMAI4 分钟前
Akamai Valkey 托管数据库:企业 AI 的实时内存解决方案
人工智能·云计算
染指11106 分钟前
118.Agent-LangChain核心组件-StructuredOutPut结构化输出-工具策略(ToolStrategy )
人工智能·langchain·agent·agents
kaixin_啊啊10 分钟前
中国研究生数学建模竞赛(华为杯)学习笔记——AI绘图方法
人工智能·笔记·学习·数学建模
夏日的盒盒14 分钟前
Science期刊下与膝关节相关的论文
人工智能·深度学习·医学·医学影像分析·膝关节
Agent工坊15 分钟前
告别「帮我写个 Prompt」:我用 LangGraph + Vue3 搭了一个本地提示词生成器 PromptOps
人工智能
桃西西呀19 分钟前
本地 RAG 问答 Agent:切片即建模,幻觉即责任
人工智能·llm·agent
wizardpisces22 分钟前
Claude Code 是 Angular,Codex 是 Vue,DeepSeek 想当 React
前端·人工智能
XianMing的博客23 分钟前
《Attention Is All You Need》学习记录(从零吃透版)
人工智能·大模型
RaaS10023 分钟前
Trae 实战:4 项功能把活整个交给 AI
人工智能