我试图在tensorflow中关注lstm-with-attention脚本: https://tensorflow.googlesource.com/tensorflow/+/master/tensorflow/models/rnn/translate
这不使用嵌入,即使它调用embedding_rnn_seq2seq,但是只有标量的单词ID,而不是任何类型的嵌入向量。
。至少我找不到嵌入参数,矩阵,也不能提供预训练嵌入的可能性。任何的想法?很多人
答案 0 :(得分:2)
translate.py
中的代码使用嵌入,但其创建深埋在实现中。该模型使用(未记录的)tf.nn.seq2seq.embedding_attention_seq2seq()
函数,creates一个tf.nn.rnn_cell.EmbeddingWrapper
在内部创建嵌入矩阵。