线性回归的框架实现和非框架实现
这是看了李沐大神的动手深度学习,自己敲了一遍,直接上代码吧,注释已经写的很清楚了
用pytorch实现
import numpy as np
import torch
from torch.utils import data
from d2l import torch as d2l
"""生成数据集"""
true_w = torch.tensor([2, -3.4])
true_b = 4.2
features, labels = d2l.synthetic_data(true_w, true_b, 1
原创
2022-01-06 11:48:54 ·
804 阅读 ·
0 评论