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

相关推荐
kyle~9 分钟前
计算机视觉---RealSense深度相机技术
人工智能·数码相机·计算机视觉·机器人·嵌入式·ros·传感器
碣石潇湘无限路1 小时前
【AI篇】当Transformer模型开始学习《孙子兵法》
人工智能·学习
看到我,请让我去学习1 小时前
OpenCV开发-初始概念
人工智能·opencv·计算机视觉
汀沿河1 小时前
8.1 prefix Tunning与Prompt Tunning模型微调方法
linux·运维·服务器·人工智能
陈敬雷-充电了么-CEO兼CTO1 小时前
大模型技术原理 - 基于Transformer的预训练语言模型
人工智能·深度学习·语言模型·自然语言处理·chatgpt·aigc·transformer
学术 学术 Fun1 小时前
✨ OpenAudio S1:影视级文本转语音与语音克隆Mac整合包
人工智能·语音识别
风铃喵游2 小时前
让大模型调用MCP服务变得超级简单
前端·人工智能
旷世奇才李先生2 小时前
Pillow 安装使用教程
深度学习·microsoft·pillow
booooooty2 小时前
基于Spring AI Alibaba的多智能体RAG应用
java·人工智能·spring·多智能体·rag·spring ai·ai alibaba
PyAIExplorer3 小时前
基于 OpenCV 的图像 ROI 切割实现
人工智能·opencv·计算机视觉