weixin_62614247的博客

原创 线性表示代码学习

create_data()生成待预测的数据，三个参数中w表示权重weight，b表示偏差bias，data_num表示数据的个数，其中torch.normal()生成随机数x，第一个参数表示生成随机数张量正态分布的平均值，第二个参数表示生成随机数张量正态分布的标准差，第三个参数表示生成张量的维度，根据生成的随机数x计算出y的值，并给该值加噪音noise以模拟真实情况。fun函数计算y的预测值，maeloss函数计算loss的平均值 ,sgd为随机梯度下降，其中参数para为梯度，lr为学习率。