Get LSTM or GRU.
cell_type (Union[RNN, str]) – “LSTM” or “GRU”
returns GRU or LSTM RNN module
Type[RNN]
previous
LSTM
next
RecurrentNetwork