标签: tensorflow
通过以下提交,tf.nn.seq2seq.embedding_attention_decoder使用tf.nn.embeddin_lookup而不使用tf.device('/cpu:0')。
tf.nn.seq2seq.embedding_attention_decoder
tf.nn.embeddin_lookup
tf.device('/cpu:0')
https://github.com/tensorflow/tensorflow/commit/8cc567bf9703e14f0e16595eb9f220859a5eff20#diff-cf94d06c60c56f954b1926fb6e53cb11L86
为什么tf.nn.seq2seq.embedding_attention_decoder不再使用tf.device('/cpu:0')? 目前是否在GPU上实现了tf.nn.embeddin_lookup?