2017-10-25 2 views
0

제 질문은 tensorflow에서 양방향 LSTM 셀 (fw_cell 및 bw_cell)을 정의하는 것입니다. fw_cell 및 bw_cell을 독립적으로 또는 동일하게 정의해야합니까?TensorFlow의 양방향 LSTM 셀

def lstm_rnn_cell(num_units, dropout): 
    _cell = tf.nn.rnn_cell.LSTMCell(num_units,state_is_tuple = True) 
    _cell = tf.contrib.rnn.DropoutWrapper(_cell, output_keep_prob = dropout) 
    return _cell 

을 다음을 수행 할 수 있습니다 :

별도로 각 셀을 정의하는 기능을 만들 수 있습니다
+0

당신은 둘 다 할 수 있습니다 – Pop

답변

4

, 당신은이 같은 것을 사용할 수 있습니다

fw_cell = tf.nn.rnn_cell.MultiRNNCell([lstm_rnn_cell(rnn_size, dropout = dropout) for _ in range(num_layers)], state_is_tuple = True) 
bw_cell = tf.nn.rnn_cell.MultiRNNCell([lstm_rnn_cell(rnn_size, dropout = dropout) for _ in range(num_layers)], state_is_tuple = True) 

을 나는 그래서 MultiRNNCell 사용이 예에서 당신은 또한 수정하여 주어진 RNN에 대한 레이어의 수를 정의 할 수 있습니다 num_layers

관련 문제