singleline_models.lstm
lstm_layer
lstm_layer (ni, nh, bidirectional=False, use_recurrent_dropout=False, r_dropout_prob=0.0, use_layer_norm=False, layer_norm_learnable=False, lstm_impl='builtin')
Creates an LSTM layer, using a different underlying LSTM implementation. - ‘builtin’ uses pytorch nn.LSTM
- ‘rnnlib’ uses LayerNormLSTM from rnnlib