【Pytorch实战(三)】线性回归

一、使用函数求解最小二乘的权重

最小二乘法是一种最基本的线性回归方法，使用torch.lstsq直接实现最小二乘法的代码如下：

import torch

x = torch.tensor([[1., 2., 1.], [2., 4., 1.], [3., 5., 1.], [4., 2., 1.], [4., 4., 1.]])
y = torch.tensor([-12., 13., 15., 14., 18.])

wr, _ = torch.lstsq(y