PyTorch如何实现多层全连接神经网络

什么是全连接神经网络

PyTorch如何实现多层全连接神经网络_第1张图片连接神经网络模型是一种多层感知机(MLP),感知机的原理是寻找类别间最合理、最具有鲁棒性的超平面,最具代表的感知机是SVM支持向量机算法。神经网络同时借鉴了感知机和仿生学,通常来说,动物神经接受一个信号后会发送各个神经元,各个神经元接受输入后根据自身判断,激活产生输出信号后汇总从而实现对信息源实现识别、分类,一个典型的神经网络如下图所示:

如何实现多层全连接神经网络

全连接神经网络按类可分为三层,分别是输入层、隐藏层,输出层;其中输入层和输出层是已经确定的,要实现多层全连接神经网络,其方法就是将隐藏层设为多层的神经网络结构。

隐层层为一层时代码实现:

PyTorch如何实现多层全连接神经网络_第2张图片

执行结果:

PyTorch如何实现多层全连接神经网络_第3张图片

隐藏层设置为3层时代码实现:

PyTorch如何实现多层全连接神经网络_第4张图片

执行结果:

PyTorch如何实现多层全连接神经网络_第5张图片

结语

通过设置隐藏层的层数,可以成功的实现多层全连接神经网络,但是需要注意的时数据的输入格式,本次我们所用的数据为784维的向量作为输入数据,所以第一层的in_features的参数为784,如果更改为其他的参数,很有可能就会报错。

你可能感兴趣的:(PyTorch图像分类教程,pytorch,神经网络,深度学习)