摘要: #模型为w1*x*x+w2*x+b import torchx_data=[1,2,3]y_data=[2,4,6]#三个tensorw1=torch.tensor([1.0])w1.requires_grad=Truew2=torch.tensor([2.0])w2.requires_grad=T 阅读全文
posted @ 2022-02-04 11:05 wxrdml 阅读(407) 评论(0) 推荐(0) 编辑
摘要: import torchx_data=[1,2,3]y_data=[2,4,6]w=torch.tensor([1.0])w.requires_grad=True #计算梯度,要特地设置,不要自动产生梯度#构建计算图def forward(x): return x*w #x and w都是tenso 阅读全文
posted @ 2022-02-04 11:00 wxrdml 阅读(64) 评论(0) 推荐(0) 编辑
摘要: #模型x*W+b,三维图象横坐标是w,纵坐标是b,竖坐标是损失函数import numpy as npimport matplotlib.pyplot as pltfrom matplotlib import cmfrom modulefinder import *from mpl_toolkits 阅读全文
posted @ 2022-02-04 10:58 wxrdml 阅读(211) 评论(0) 推荐(0) 编辑