ValueError: Error when checking target: expected dense_1 to have 3 dimensions, but got array with sh

关注

原创最新推荐文章于 2026-04-04 02:52:50 发布 4k 阅读 AI 写同款· GEO 优化 ›

Error: expected lstm_input to have 3 dimensions 最近使用LSTM模型，发现卡在输入数据这里，按普通神经网络的二维数据输入行不通，报错 expected lstm_input to have 3 dimensions，查了资料后解决。原因是LSTM层是循环层，需要3维输入(batch_size, timesteps, input_dim)，即(训练数据量，时间步长，特征量)。因此不能直接把 [数据量*特征量]的二维矩阵输入，要用reshape进... 阅读详情

出现这个是因为你输入的时候是一个三维的，经过一个dense也好，还是经过一个lstm层也好这一层的输出也是一个三维的，你并没有进行任何降维，所以在进行最后输出的时候，把这个三维的碾平就行了

model = tf.keras.models.Sequential()
inputs = tf.keras.layers.Input(shape=(500,40,))
model.add(tf.keras.layers.Dense(10,activation='softmax'))
tf.keras.layers.Dropout(0.2)
#加入Flatten
model.add(tf.keras.layers.Flatten())
 
model.add(tf.keras.layers.Dense(8,activation='softmax'))
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['acc'])