每天五分钟深度学习pytorch：基于pytorch搭建普通全连接神经网络

本文重点

本文我们通过pytorch搭建普通的全连接神经网络，这里我们就不介绍什么是全连接神经网络了，如果不知道的可以看我的机器学习专栏，或者深度学习专栏，它们对全连接神经网络都进行了简单的介绍。

代码

import torch
from torch import nn
class ThreeNet(nn.Module) : 
 def __init__ (self,in_dim,n_hidden_1,n_hidden_2,out_dim):
 super(simpleNet, self).__init__()
 self.layer1 = nn.Linear (in_dim, n_hidden_1) 
 self.layer2 = nn.Linear(n_hidden_1,n_hidden_2) 
 self.layer3 = nn.Linear(n_hidden_2, out_dim)
 def forward(self,x): 
 x =self.layer1(x) 
 x =self.layer2(x) 
 x =self.layer3(x) 
 return x

上面的代码是建立普通的三层的全连接神经网络，层与层之间没有激活函数，其中__init__࿰