keras:LSTM部分參數解析

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
    在這裏插入圖片描述
    上面的圖中,每個綠色矩形框是一個時間步,對應序列中的一個元素,hth_t是一個向量,其長度就是lstm的超參數。
發佈了20 篇原創文章 · 獲贊 7 · 訪問量 5615
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章