用图解释RNN程序的部分过程

LSTM的隐藏层的单元个数为hidden_size,因此,LSTM每一步的输出数据的维度为(batch_size,hidden_size)。有因为LSTM展开的时间步数为num_steps,于是通过 outputs.append(cell_output) 将每一时刻的输出都收集起来,这样,最后的outputs是一个list。 整个过程为: 图中黄色的部分表示同一个序列在LSTM不同时刻的输出。 紧
相关文章
相关标签/搜索