什麼是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實現將運行得更快。