RNN — PyTorch 1.10.1 documentation
pytorch.org › docs › stableRNN — PyTorch 1.10.0 documentation RNN class torch.nn.RNN(*args, **kwargs) [source] Applies a multi-layer Elman RNN with \tanh tanh or \text {ReLU} ReLU non-linearity to an input sequence. For each element in the input sequence, each layer computes the following function: h_t = \tanh (W_ {ih} x_t + b_ {ih} + W_ {hh} h_ { (t-1)} + b_ {hh}) ht