RuntimeError:——Expected dimensional不匹配

**

RuntimeError: Expected 4-dimensional input for 4-dimensional weight [10, 1, 5, 5], but got 3-dimensional input of size [1, 28, 28] instead

**
在这里插入图片描述
网络的第一层
RuntimeError:——Expected dimensional不匹配_第1张图片
此时我处理的是一张图片,而卷积神经网络模型是输入的是一个数据集的图片,而不是一张图片。
此时我的输入维度如下:
RuntimeError:——Expected dimensional不匹配_第2张图片
可通过如下方式来增加输入图片的维度:
在这里插入图片描述
增加维度过后:
RuntimeError:——Expected dimensional不匹配_第3张图片

至此,问题解决。
附注:最开始我以为这个报错的原因是:输入图片的数据大小应该和网络第一层设计的尺寸大小相同,就拿这个例子距离,我误以为错误是输入的图片[1,28,28]中12828不等于网络第一层的1015*5([10,1,5,5])所造成的,其实是维度相同就可以了,是自己对神经网络结构理解得不够。

你可能感兴趣的:(神经网络,神经网络,python,深度学习)