LSTM中模型的输出为ht,中间状态为ct
it=σ(Wiixt+bii+Whih(t−1)+bhi)ft=σ(Wifxt+bif+Whfh(t−1)+bhf)gt=tanh(Wigxt+big+Whgh(t−1)+bhg)ot=σ(Wioxt+bio+Whoh(t−1)+bho)ct=ft∗c(t−1)+it∗gtht=ot∗tanh(ct)
由ht=ot∗tanh(ct)可知,ht为LSTM模型的输出,ct为中间状态。
pytorch LSTM输出,out表示序列所有时刻的输出,而hn表示最后一个时刻的输出
![pytorch中LSTM参数的理解 pytorch中LSTM参数的理解](/default/index/img?u=aHR0cHM6Ly9waWFuc2hlbi5jb20vaW1hZ2VzLzMwOC9jMjFhMjk1OTViNDdjMTQ2N2ZiZjI2MzRiYjc1MDkwYy5wbmc=)