使用 pytorch 创建神经网络拟合sin函数

本博客写一个小例子,使用 pytorch 来编写一个神经网络来拟合 sin 函数 废话少说,直接上代码: from torch.utils.data import DataLoader from torch.utils.data import TensorDataset import torch.nn as nn import numpy as np import torch # 准备数据 x=np.linspace(-2*np.pi,2*np.pi,400) y=np.sin(x) # 将数据做成数据集的模样 X=np.expand_dims(x,axis=1) Y=y.reshape(40