LSTM需要學習的參數個數

 結論先行: lstm所需學習的參數 = 4 * [(  input_size + hidden_num )*  hidden_num + hidden_num ] 我們知道輸入一個28維的向量,會映射到128維來進行表示, 輸出ht和記憶ct維度相同。 下面我們假設有13*5大小的圖片,放進LSTM,隱含層神經元數量爲10   ht-1 是上一時刻的輸出,維數與隱含層神經元數量一致:1*10 x
相關文章
相關標籤/搜索