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

相关推荐
LaughingZhu1 分钟前
Product Hunt 每日热榜 | 2026-07-05
人工智能·经验分享·深度学习·神经网络·产品运营
海兰9 分钟前
【AI编程思考:第六篇】学习记忆与 Agent 状态:让 AI 真正“记得”你
人工智能·学习·ai编程
Dragon Wu13 分钟前
ComfyUI ReActor需要下载检测模型的解决方案
人工智能·ai
雨辰AI14 分钟前
【零基础实战】大模型入门面试 100 问:基础概念 + 环境实操(一问一答版,直接背诵)
人工智能·ai·面试·职场和发展·ai编程
龙萱坤诺15 分钟前
Claude Fable 5 重新开放:最强模型回归
大数据·运维·人工智能
月下的树15 分钟前
双目视觉深度原理
人工智能
Token炼金师16 分钟前
数据工程层:企业AI项目七成折戟的第一道死亡谷
人工智能·深度学习·llm
Sam092729 分钟前
【AI 算法精讲 16】BPE 分词:从字节对到子词
人工智能·python·算法·ai
龙虾PRO35 分钟前
打通OpenClaw与Claude Code会话链路,实现AI全自动编码落地
大数据·人工智能
余俊晖39 分钟前
多模态文档解析开源新进展:Unlimited OCR技术方案
人工智能·ocr·多模态