torch.nn.LSTMCell类是一个LSTM的一个cell。数学表达式为:
i=σ(Wiix+bii+Whih+bhi)f=σ(Wifx+bif+Whfh+bhf)g=tanh(Wigx+big+Whgh+bhg)o=σ(Wiox+bio+Whoh+bho)c′=f∗c+i∗gh′=o∗tanh(c′)
其中,σ是一个sigmoid函数,∗是Hadamard积。
LSTM的cell如下图所示:

相关文章:
-
2021-08-07
-
2021-12-05
-
2021-12-05
-
2021-05-23
-
2021-05-09
-
2021-04-01
-
2021-04-01
-
2021-12-28