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=fc+igh=otanh(c) \begin{array}{ll} i = \sigma(W_{ii} x + b_{ii} + W_{hi} h + b_{hi}) \\ f = \sigma(W_{if} x + b_{if} + W_{hf} h + b_{hf}) \\ g = \tanh(W_{ig} x + b_{ig} + W_{hg} h + b_{hg}) \\ o = \sigma(W_{io} x + b_{io} + W_{ho} h + b_{ho}) \\ c' = f * c + i * g \\ h' = o * \tanh(c') \\ \end{array}
其中,σ\sigma是一个sigmoid函数,*是Hadamard积。

LSTM的cell如下图所示:
PyTorch系列:torch.nn.LSTMCell

相关文章:

  • 2021-08-07
  • 2021-12-05
  • 2021-12-05
  • 2021-05-23
  • 2021-05-09
  • 2021-04-01
  • 2021-04-01
  • 2021-12-28
猜你喜欢
  • 2021-07-21
  • 2021-08-17
  • 2021-12-11
  • 2021-05-13
  • 2021-11-10
  • 2021-09-07
相关资源
相似解决方案