Pytorch学习笔记--Bilibili刘二大人Pytorch教学代码汇总(注释及运行结果)
Part103Gradient_Descent1# 输入训练数据x_data = [1.0, 2.0, 3.0]y_data = [2.0, 4.0, 6.0]# 设置初始参数w = 1.0 # 初始权重alpha = 0.005 #初始梯度下降法的学习率# 定义计算y_hat的函数def forward(x): return x * w# 定义计算平均损失的函数def cost(xs, ys): sum_cost = 0 for x, y in z.