说明:一维线性拟合
数据为y=4x+5加上噪音
结果:
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import pyplot as plt
from torch.autograd import Variable
import torch
from torch import nn
X = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
Y <weixin_38685694> 上传 | 大小:49kb