2021
03-24
03-24
pytorch动态神经网络(拟合)实现
(1)首先要建立数据集importtorch#引用torch模块importmatplotlib.pyplotasplt#引用画图模块x=torch.unsqueeze(torch.linspace(-1,1,100),dim=1)#产生(-1,1)的100个点横坐标,dim表示维度,表示在这里增加第二维y=x.pow(2)+0.2*torch.rand(x,size())#0.2*torch.rand(x,size())是为了产生噪点使数据更加真实(2)建立神经网络importtorchimoorttorch.nn.functionalasF#激励函数在这个模块里classNet(torch.nn.Modul...
继续阅读 >