singleline_models.lstm


source

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