Building an LSTM model under Tensorflow