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

相关推荐
qq_452396232 分钟前
第二篇:《安装、配置与 psql 高效使用》
人工智能
方方洛8 分钟前
ray教程-11-可观测性与调试
人工智能·分布式·机器学习
我是李工啊11 分钟前
PRD 总和代码对不上?让 AI 从代码生成状态图和权限矩阵
人工智能·产品经理
ComputerInBook13 分钟前
泛函分析概观
人工智能·算法·机器学习·泛函分析
burg_xun24 分钟前
【生命科学 Agent】推理平台(二)
人工智能·程序人生·系统架构
IT_陈寒32 分钟前
Vue的这个响应式陷阱我居然现在才发现
前端·人工智能·后端
lanxiaoqing0234 分钟前
演讲训练的第一步:先敢说,再会说
人工智能·学习·学习方法·口才
巫山老妖35 分钟前
REA 工程解析:让 AI Agent 拿着证据做逆向
人工智能
梦想的颜色35 分钟前
【AI速览】Codex 系列模型选型指南:效率、成本、任务匹配全维度决策
人工智能·chatgpt·codex·luna·astra·sol·大模型选型
安全指北针1 小时前
密态计算:数据 “可用不可见” 的底层技术解析
人工智能·算法