tensorflow tf.nn.embedding_lookup(embeddings, train_inputs)解释

来源:互联网 发布:c语言指针详解 编辑:程序博客网 时间:2024/04/28 23:11

embedding_lookup从表面意思看就是一个查找单词对应词向量的表,
embeddings参数:词跟词向量的对应表
train_inputs参数:需要转换的文本(int类型)
类似下面的操作:

embeddings = np.random.random([1024, 64])  # 64-dimensional embeddingstrain_inputs = np.array([0, 5, 17, 33])return matrix[train_inputs]  # return a matrix of shape [4, 64] 
0 0
原创粉丝点击