PyTorch深度学习实践（四）

写在前面：学习用pytorch的工具，来重现之前学习的线性模型！

1 pytorch的功能

最后得到的loss应该是一个标量，而不是一个向量。

2 构造模型——线性模型

• Linear包含了权重和输入
• 用Module定义的模型，会自动反向传播

3 代码

• 可以直接运行

import torch

x_data = torch.Tensor([[1.0], [2.0], [3.0]])
y_data = torch.Tensor([<