「@Author:Runsen」
在PyTorch建立模型,主要是NN模块。
nn.Linear
nn.Linear是创建一个线性层。这里需要将输入和输出维度作为参数传递。...和tensorflow中的Sequential完全一样。...0.0000, 0.8861],
[1.0895, 0.0000],
[1.3047, 0.0000]], grad_fn=)
在上面的模型中缺少了优化器...,训练线性模型得到loss为0.7467,上面就是NN模型建立model的整个流程,
第一个神经网络模型
下面实现第一个分类神经网络,其中一个隐藏层用于开发单个输出单元。...使用下面代码,创建一个顺序模型 -
model = nn.Sequential(nn.Linear(n_in, n_h),
nn.ReLU(),
nn.Linear(n_h, n_out),