注:本代码在jupyter notebook上运行
封面图片来源
线性回归的简洁实现
1、生成数据集
import numpy as np
import torch
from torch.utils import data
from d2l import torch as d2l
def synthetic_data(w, b, num_examples): #@save
"""生成y=Xw+b+噪声"""
X = torch.normal(0, 1, (num_examples, len(w))) # [1000,2]
y = torch.matmul(X, w) + b # [1000,2]*[2, 1]=[1000, 1]
y += torch.normal(0, 0.01, y.shape)