Какова функция активации по умолчанию для cudnnlstm в tenorflow - PullRequest
0 голосов
/ 25 октября 2018

Какая функция активации по умолчанию cudnnlstm в TensorFlow?Как я могу установить функцию активации, такую ​​как relu?Может это просто линейная модель?Я прочитал документ, но не нашел его.

Например, код ниже:

lstmcell=tf.contrib.cudnn_rnn.CudnnLSTM(1,encoder_size,direction="bidirectional")

hq,_ =lstmcell(query)

И я прочитал документ TensorFlow По этой ссылке *функция cudnnlstm в TensorFlow и как ее изменить на leaky_relu.

1 Ответ

0 голосов
/ 02 апреля 2019
tf.contrib.cudnn_rnn.CudnnLSTM() : Tanh

Это было дано в github Keras.

https://github.com/keras-team/keras/issues/8510#issuecomment-429255318

Документация Nvidia.

https://devblogs.nvidia.com/optimizing-recurrent-neural-networks-cudnn-5/

...