tf.nn.embedding_lookup()就是根据input_ids中的id,寻找embeddings中的第id行。
比如input_ids=[1,3,5],则找出embeddings中第1,3,5行,组成一个tensor返回。
embedding_lookup不是简单的查表,id对应的向量是可以训练的
tf.nn.embedding_lookup()就是根据input_ids中的id,寻找embeddings中的第id行。
比如input_ids=[1,3,5],则找出embeddings中第1,3,5行,组成一个tensor返回。
embedding_lookup不是简单的查表,id对应的向量是可以训练的