原文链接
embedding_lookup()的用法
这个函数真的很常用,尤其word2vec
tf.nn.embedding_lookup()就是根据input_ids中的id,寻找embeddings中的第id行。比如input_ids=[1,3,5],则找出embeddings中第1,3,5行,组成一个tensor返回。
原文链接
embedding_lookup()的用法
这个函数真的很常用,尤其word2vec
tf.nn.embedding_lookup()就是根据input_ids中的id,寻找embeddings中的第id行。比如input_ids=[1,3,5],则找出embeddings中第1,3,5行,组成一个tensor返回。