正确实现tf.contrib.cudnn_rnn.CudnnRNNRelu()

时间:2019-04-06 08:03:19

标签: tensorflow

我正在尝试生成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]

但这似乎不起作用。

对于CudnnCompatibleLSTMCellCudnnLSTMtf.contribl.cudnn_rnn类之间的区别的解释也将不胜感激。

0 个答案:

没有答案