关于LSTM层与全连接层的接口逻辑及代码结构可视化咨询

代码逻辑说明 #

你给出的Keras代码：

lstm4 = LSTM(3, activation='relu')(lstm3) 
DEN = Dense(4)(lstm4)  

在Keras框架中，LSTM层默认参数return_sequences=False，这意味着该层只会输出最后一个时间步的隐藏状态，而非所有时间步的输出。

与全连接层的协作逻辑 #

仅最后一个LSTM细胞的最终输出（一个维度为3的向量）会接入全连接层（Dense(4)）的每个神经元，全连接层的每个神经元都会接收这个向量的全部元素进行计算。

对应示意图判断 #

你提供的示意图中：

• 图1：所有LSTM细胞的输出都接入全连接层
• 图2：每个LSTM细胞输出既传入下一时间步LSTM细胞，又接入全连接层

这段代码的逻辑不符合上述任一图，正确的结构应该是：

• 输入序列按时间步依次流入LSTM的各个细胞，每个细胞的输出仅传递给下一个时间步的LSTM细胞
• 只有序列最后一个时间步的LSTM细胞输出，作为全连接层的唯一输入，进入全连接层完成后续计算

内容的提问来源于stack exchange，提问作者Steven Dascoli