pytorch nn.LSTM()，nn.GRU()參數詳解

LSTM結構中是一個神經網絡，即上圖的結構就是一個LSTM單元，裏面的每一個黃框是一個神經網絡，這個網絡的隱藏單元個數咱們設爲hidden_size，那麼這個LSTM單元裏就有4*hidden_size個參數。每一個LSTM輸出的都是向量，包括 C t 和 h t C_t和h_t Ct​和ht​ ，它們的長度都是當前LSTM單元的hidden_sizehtml 函數 class torch.nn.

>>阅读原文<<