用圖解釋RNN程序的部分過程

LSTM的隱藏層的單元個數爲hidden_size,因此,LSTM每一步的輸出數據的維度爲(batch_size,hidden_size)。有因爲LSTM展開的時間步數爲num_steps,於是通過 outputs.append(cell_output) 將每一時刻的輸出都收集起來,這樣,最後的outputs是一個list。 整個過程爲: 圖中黃色的部分表示同一個序列在LSTM不同時刻的輸出。 緊
相關文章
相關標籤/搜索