我正在尝试生成Cudnn优化的单层RNN模型的实现。我找不到很好的实现,我想知道我是否可以就我在这里做错的事情获得投入?
基于我对张量流docstring的阅读
# Assume tensor_4d is a 4d input tensor
input_tensor = tf.unstack(tensor_4d, axis=1)
single_cell = lambda: tf.contrib.cudnn_rnn.CudnnRNNRelu(num_units)
cell = tf.nn.rnn_cell.MultiRNNCell(
[single_cell() for _ in range(num_layers)])
outputs, final_state = tf.nn.static_rnn(cell, inputs, dtype=tf.float32)
final_out = outputs[-1]
但这似乎不起作用。
对于CudnnCompatibleLSTMCell
中CudnnLSTM
和tf.contribl.cudnn_rnn
类之间的区别的解释也将不胜感激。