Input 0 of layer dense is incompatible with the layer: expected axis -1 of input shape to have value

最新推荐文章于 2024-03-08 19:11:19 发布

fehanh

最新推荐文章于 2024-03-08 19:11:19 发布

阅读量6.5k

点赞数 3

文章标签：机器学习

本文链接：https://blog.csdn.net/weixin_45139997/article/details/121562473

版权

使用Keras的Sequential框架搭建神经网络模型，在使用模型分类时报错：

Input 0 of layer dense is incompatible with the layer: expected axis -1 of input shape to have value 784 but received input with shape [None, 28]

报错部分代码：

    model.fit(train_image, train_label_onehot, epochs=5)  
    print(model.predict(train_image[0]))

分析报错原因：

报错内容的大致意思为：第0层全连接层的输入与该层不兼容（该模型的第0层采用的Flatten层），期望输入的最后那个维度的形状有784个变量，但是实际接收到的输入的形状是 [None, 28]

分析：很明显，这个错误是维度上的问题。
所以在train_image[0]再添加一个中括号，改为train_image[[0]]
各表达的形状：

    print(train_image.shape)
    print(train_image[0].shape)
    print(train_image[[0]].shape)

输出为

(60000, 28, 28)
(28, 28)
(1, 28, 28)

从结果中可以看出，从测试集本身有三个维度，从测试集中取出一个样本，变成了两个维度，所以才会报错。

针对这个相似的报错的问题，基本只有两种原因：

①维度不匹配
②维度匹配但这一维度的变量数不匹配，导致无法输入到模型中去

关注