pytorch中LSTM參數的理解

LSTM中模型的輸出爲ht,中間狀態爲ct i t = σ ( W i i x t + b i i + W h i h ( t − 1 ) + b h i ) f t = σ ( W i f x t + b i f + W h f h ( t − 1 ) + b h f ) g t = tanh ⁡ ( W i g x t + b i g + W h g h ( t − 1 ) + b h g )

>>阅读原文<<