Pytorch nn.Linear()

nn.Linear就是神经网络中的线性层,类似于数学中的线性函数,可以实现形如y=X*weight^T+b的功能。

python 复制代码
#导包
import torch.nn as nn
import torch

#创建1个张量
sample=torch.tensor([1.,10.,100.])

#nn.Linear(in_feature,out_feature,bias),这里设置了不需要bias,即函数为y=k*x

linear=nn.Linear(3,3,bias=False)

print(linear.weight)

#weight为一个3*3的张量
out:
Parameter containing:
tensor([[ 0.0777,  0.1295, -0.3284],
        [-0.5325,  0.2380,  0.1290],
        [ 0.3780, -0.1113,  0.3035]], requires_grad=True)

output=linear(sample)

print(output)
out:
tensor([-31.4626,  14.7472,  29.6170], grad_fn=<SqueezeBackward3>)

我们输入有3个特征 x1,x2,x3,bias=False,所以方程式为y=w1*x1+w2*x2+w3*x3

y1=0.0777*1+0.1295*10+(-0.3284)*100=-31.46

y2=-0.5325*1+0.2380*10+0.1290*100=14.74

y3=0.3780*1+(-0.1113)*10+0.3035*100=29.61

输出特征也为3

相关推荐
AI Echoes3 小时前
LLMOps平台:开源项目LMForge = GPTs + Coze
人工智能·python·langchain·开源·agent
风信子的猫Redamancy3 小时前
文心大模型 X1.1:百度交出的“新深度思考”答卷
人工智能·百度·大模型·深度思考
聚客AI3 小时前
🚀从零构建AI智能体:九大核心技术拆解与落地建议
人工智能·agent·mcp
HUIMU_3 小时前
YOLOv5实战-GPU版本的pytorch虚拟环境配置
人工智能·pytorch·深度学习·yolo
虚行3 小时前
VisionMaster - 1.图像源
人工智能·计算机视觉
Coovally AI模型快速验证3 小时前
基于YOLO集成模型的无人机多光谱风电部件缺陷检测
人工智能·安全·yolo·目标跟踪·无人机
猫天意3 小时前
【CVPR2023】奔跑而非行走:追求更高FLOPS以实现更快神经网络
人工智能·深度学习·神经网络·算法·机器学习·卷积神经网络
1373i3 小时前
【Python】pytorch数据操作
开发语言·pytorch·python
杀生丸学AI3 小时前
【三维重建】3R-GS:优化相机位姿的3DGS最佳实践
人工智能·3d·aigc·三维重建·视觉大模型·高斯泼溅
羊羊小栈3 小时前
基于「YOLO目标检测 + 多模态AI分析」的PCB缺陷检测分析系统(vue+flask+数据集+模型训练)
vue.js·人工智能·yolo·目标检测·flask·毕业设计·大作业