get_rnn# pytorch_forecasting.models.nn.rnn.get_rnn(cell_type: Type[RNN] | str) → Type[RNN][source]# Get LSTM or GRU. Parameters: cell_type (Union[RNN, str]) – “LSTM” or “GRU” Returns: returns GRU or LSTM RNN module Return type: Type[RNN]
