torch中的实现简单的反向传播
#加油加油加油#认真对待每一行代码#努力写好每一行代码#搞懂每一行代码import torchx_data = [1.0, 2.0, 3.0]y_data = [2.0, 4.0, 6.0]w = torch.Tensor([1.0])w.requires_grad = True"""因为torch.Tensor函数,所以我们的w是一个张量;下面我们计算时,返回的也是张量;这样就可以有一其他功能,便可直接函数取出我们想要的值;item()是从原来的精度上更加精确;最外的两个.
原创
2022-05-28 15:31:34 ·
1231 阅读 ·
0 评论