什么是Tensorflow中的rnn单元的cuDNN实施方案

Question

要创建RNN单元，有几个类如GRUCell和LSTMCell，它们可以稍后用于创建RNN层。
还有另外两个类CudnnGRU和CudnnLSTM可以直接用于创建RNN图层。

在文档中他们说后者类有cuDNN的实现。当我创建RNN模型时，为什么我应该使用或不使用这个cuDNN已实现的类而不是经典的RNN实现？

Answer 1

简而言之：cudnnGRU和cudnnLSTM可以/必须在GPU上使用，而不是常规的rnn实现。所以如果你有tensorflow-gpu，那么RNN单元的cudnn实现将运行得更快。