experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove unnecessary tf.exp

Browse Source
This commit is contained in:
Ruizhi 2018-08-14 10:28:06 +08:00 committed by GitHub
parent 33c1014549
commit fee2c48d3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/eager/python/examples/generative_examples/image_captioning_with_attention.ipynb
View File

@ -1056,7 +1056,7 @@
"\n", "\n",
" attention_plot[i] = tf.reshape(attention_weights, (-1, )).numpy()\n", " attention_plot[i] = tf.reshape(attention_weights, (-1, )).numpy()\n",
"\n", "\n",
" predicted_id = tf.multinomial(tf.exp(predictions), num_samples=1)[0][0].numpy()\n", " predicted_id = tf.multinomial(predictions, num_samples=1)[0][0].numpy()\n",
" result.append(index_word[predicted_id])\n", " result.append(index_word[predicted_id])\n",
"\n", "\n",
" if index_word[predicted_id] == '<end>':\n", " if index_word[predicted_id] == '<end>':\n",