我正在尝试在 tensorflow 中构建一个序列到序列模型,并且我正在使用 MultiRNN 类为我的编码器将 3 个 GRU 链接在一起。
with tf.variable_scope('encoder') as encode_scope:
encoder_rnn = rnn.MultiRNNCell([rnn.GRUCell(1024)] * 3)
encoder_state = tf.zeros((BATCH_SIZE, encoder_rnn.state_size))
for t in range(TIME_STEPS):
encoder_output, encoder_state = encoder_rnn(getTimeStep(input_tensor, t), encoder_state)
encode_scope.reuse_variables()
我目前正在张量板上查看我的模型。当我在图中展开编码器节点时,它看起来像这样......
它继续向左延伸,直到到达单元格 63。这些单元格到底是什么,为什么有 63 个?当我扩展编码器节点(每个 GRU 1 个)时,我期待看到更像 3 个项目。