wo_squirrel的博客

在给定线性模型： 给定数据： x: 1.0 2.0 3.0 y:2.0 4.0 6.0 下利用反向传播算法得到三个参数。 import torch x_data = [1.0, 2.0, 3.0] y_data = [2.0, 4.0, 6.0] w1 = torch.Tensor([1.0]) w2 = torch.Tensor([1.0]) b = torch.Tensor([1.0]) w1.requires_grad = True w2.requires_grad = T