PyTorch不会将直线拟合到两个数据点

import torch
import torch.nn as nn
import torch.optim as optim

# 定义数据
x = torch.tensor([1.0, 2.0])
y = torch.tensor([3.0, 4.0])

# 定义模型
model = nn.Linear(1, 1)  # 线性模型，输入维度为1，输出维度为1

# 定义损失函数和优化器
criterion = nn.MSELoss()  # 均方误差损失函数
optimizer = optim.SGD(model.parameters(), lr=0.01)  # 随机梯度下降优化器

# 训练模型
for epoch in range(100):
    optimizer.zero_grad()  # 梯度清零
    outputs = model(x.unsqueeze(1))  # 模型预测
    loss = criterion(outputs.squeeze(), y)  # 计算损失
    loss.backward()  # 反向传播
    optimizer.step()  # 更新参数

# 打印训练结果
print(model.weight)  # 输出拟合的直线斜率
print(model.bias)  # 输出拟合的直线截距