LSTM — PyTorch 1.10.1 documentation
pytorch.org › docs › stableLSTM. class torch.nn.LSTM(*args, **kwargs) [source] Applies a multi-layer long short-term memory (LSTM) RNN to an input sequence. For each element in the input sequence, each layer computes the following function: 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 ...