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

相关推荐
新知图书3 分钟前
测试与发布(新闻早报智能体开发)
人工智能·agent·ai agent·智能体·扣子
lisw055 分钟前
【计算机科学技术】从晶圆制备到芯片出厂的流程
人工智能·机器学习·制造
武子康6 分钟前
OpenAI 为什么需要 FDE:模型公司正在变成交付公司
人工智能·llm·openai
alxraves9 分钟前
零成本搭建个人AI Agent:在本地电脑上部署免费智能助手详细教程
人工智能·电脑
AI星桥小王子21 分钟前
文字转视频 vs 图像转视频:两类 AI 生成技术效果与效率横向测评
人工智能
Bode_200225 分钟前
中台多模态异构跨库缝合与动态机理图谱自动创成方法
人工智能·智能工厂·工业大模型
哥本哈士奇30 分钟前
医疗器械行业 Salesforce Territory 完整落地实例
大数据·人工智能
念雨思39 分钟前
HarmonyOS AI 应用开发实战:猫咪行为解读 —— AI 让宠物沟通更简单
人工智能·宠物
! 冰封雪莲 !1 小时前
地表水自动监测配套采样设备怎么选?Smart WQS2000 智能采样站介绍
人工智能·环保
zandy10111 小时前
企业级BI平台选型指南:评估框架与核心能力矩阵
大数据·人工智能·矩阵