VanillaRNN, LSTM, and GRU classes are coded in this link. I\'m trying to customize the LSTM class here for my own problem but the dropout function is not added in this lstm clas