translate.py不使用嵌入?

时间:2016-02-19 15:03:09

标签: tensorflow

我试图在tensorflow中关注lstm-with-attention脚本: https://tensorflow.googlesource.com/tensorflow/+/master/tensorflow/models/rnn/translate

这不使用嵌入,即使它调用embedding_rnn_seq2seq,但是只有标量的单词ID,而不是任何类型的嵌入向量。

至少我找不到嵌入参数,矩阵,也不能提供预训练嵌入的可能性。任何的想法?很多人

1 个答案:

答案 0 :(得分:2)

translate.py 中的代码使用嵌入,但其创建深埋在实现中。该模型使用(未记录的)tf.nn.seq2seq.embedding_attention_seq2seq()函数,creates一个tf.nn.rnn_cell.EmbeddingWrapper在内部创建嵌入矩阵。