1.units: Positive integer, dimensionality of the output space.即输出维度大小
2.return_sequences
参考:https://blog.youkuaiyun.com/u011327333/article/details/78501054
源码注释如下:
即该参数为True的时候将返回整个序列的输出,否则只返回最后一个输出。以文本任务,假设LSTM有10个单元,那么返回参数为True时将返回这10个单元各自的隐藏状态(这里的隐藏状态为一个向量),适用于序列标注任务。否则只返回最后一个单元的隐藏状态,适用于序列分类任务。
输入输出
- 输入:三维,分别为batch_size, seq_length, vector_dim
- 输出:二维:seq_length, unit_num
上面的图中,每个绿色矩形框是一个时间步,对应序列中的一个元素, h t h_t ht是一个向量,其长度就是lstm的超参数。