一、视频课:刘二大人05.用PyTorch实现线性回归_哔哩哔哩_bilibili
二、
__init__函数及self的理解参考链接:
Python中 __init__函数以及参数self怎么理解和使用? - 知乎
推荐一篇大佬的笔记,真的非常简洁易懂!
PyTorch学习(四)--用PyTorch实现线性回归_陈同学爱吃方便面的博客-CSDN博客
三、
参考了上述大佬的代码并加上了画图部分
import torch
import matplotlib.pyplot as plt
x_data = torch.tensor([[1.0], [2.0], [3.0]])
y_data = torch.tensor([[2.0], [4.0], [6.0]])
class LinearModel(torch.nn.Module):
def __init__(self):