1.units: Positive integer, dimensionality of the output space.即輸出維度大小
2.return_sequences
參考:https://blog.csdn.net/u011327333/article/details/78501054
源碼註釋如下:
即該參數爲True的時候將返回整個序列的輸出,否則只返回最後一個輸出。以文本任務,假設LSTM有10個單元,那麼返回參數爲True時將返回這10個單元各自的隱藏狀態(這裏的隱藏狀態爲一個向量),適用於序列標註任務。否則只返回最後一個單元的隱藏狀態,適用於序列分類任務。
輸入輸出
- 輸入:三維,分別爲batch_size, seq_length, vector_dim
- 輸出:二維:seq_length, unit_num
上面的圖中,每個綠色矩形框是一個時間步,對應序列中的一個元素,是一個向量,其長度就是lstm的超參數。