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

相关推荐
IT_陈寒几秒前
Vite静态资源路径这个大坑害我调了一下午
前端·人工智能·后端
武哥聊编程1 分钟前
【AI实战项目】基于SpringAI+Springboot+Vue的AI面试刷题训练平台
vue.js·人工智能·spring boot·springai
摆烂工程师2 分钟前
别只拿 GPT-6 Astra 聊天,它真正恐怖的是开始会“干活”了
人工智能·程序员·vibecoding
LearnYard3 分钟前
技术博主实测:2026年大语言模型辅助学习工具横向对比
人工智能·学习·语言模型
blues925710 分钟前
赋能企业AI高效落地|Vantage万极FDE四天实战营杭州首班开启报名
大数据·人工智能·fde
深海鱼在掘金11 分钟前
深入浅出RAG——第10章:检索后处理与重排序
人工智能
阿里云大数据AI技术28 分钟前
使用 PAI,一键拉起云端 AI “投资智囊团”
人工智能·agent
四六的六32 分钟前
让 AI 自己去点后台页面,它把我们的库存点没了
前端·人工智能·agent·个人开发·ai编程·ai产品·ai前端
月华路36 分钟前
《模型不玄学》第30章 完整跑一遍 Cheat Sheet
人工智能·深度学习·机器学习
workflower39 分钟前
智能体-企业行政问答助手
人工智能·机器学习·机器人·云计算·无人机