LSTM weight noise implementation bug fix