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

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

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

  • 6
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值