本博客写一个小例子,使用 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)
# 将数据