2017-10-16 15 views
0

RNNセルを作成するには、GRUCellおよびLSTMCellのようなクラスがあり、後でRNNレイヤを作成するために使用できます。
また、CudnnGRUCudnnLSTMという2つのクラスがあり、それらを直接使用してRNNのレイヤーを作成できます。TensorflowでrnnセルのcuDNN実装とは何ですか

ドキュメントでは、後者のクラスにはcuDNNの実装があります。 RNNモデルを作成するときに、このcuDNNの実装クラスを古典的なRNNの実装に使用する必要があるのはなぜですか?

答えて

0

要約:cudnnGRUとcudnnLSTMはGPUでは使用できますが、通常の実装では使用できません。だから、もしあなたがtensorflow-gpuを持っていれば、RNN細胞のcudnn実装がより速く実行されます。

関連する問題