2017-10-16 5 views
0

ist RNN Zellen zu erstellen, gibt es Klassen, wie GRUCell und LSTMCell die später RNN Schichten erstellen verwendet werden kann.
Und es gibt auch 2 andere Klassen wie CudnnGRU und CudnnLSTM, die direkt verwendet werden können, um RNN Schichten zu erstellen.Was cuDNN Implementierung von rnn Zellen in Tensorflow

In der Dokumentation sagen sie, dass die letzteren Klassen cuDNN Implementierung haben. Warum sollte ich diese cuDNN implementierten Klassen über klassische RNN Implementierungen verwenden oder nicht verwenden, wenn ich ein RNN Modell erstelle ..?

Antwort

0

Kurz gesagt: cudnnGRU und cudnnLSTM können/müssen auf GPU verwendet werden, normale rnn-Implementierungen nicht. Wenn Sie also Tensorflow-GPU haben, würde die Implementierung von RNN-Zellen schneller laufen.