PyTorch系列:torch.nn.LSTMCell

torch.nn.LSTMCell類是一個LSTM的一個cell。數學表達式爲: i = σ ( W i i x + b i i + W h i h + b h i ) f = σ ( W i f x + b i f + W h f h + b h f ) g = tanh ⁡ ( W i g x + b i g + W h g h + b h g ) o = σ ( W i o x + b i
相關文章
相關標籤/搜索