tf.nn.embedding_lookup（）函数

最新推荐文章于 2021-04-27 10:57:43 发布

蹦跶的小羊羔

最新推荐文章于 2021-04-27 10:57:43 发布

阅读量844

点赞数

分类专栏： Tensorflow 文章标签： tensorflow

本文链接：https://blog.csdn.net/yql_617540298/article/details/88394120

版权

本文介绍了tf.nn.embedding_lookup函数的使用，该函数用于根据输入张量id选取相应索引的元素。通过示例代码说明了如何利用此函数从embedding张量中获取指定id的向量，并简单提及了Embedding原理。

摘要由CSDN通过智能技术生成

一、tf.nn.embedding_lookup（）

tf.nn.embedding_lookup函数的用法主要是选取一个张量里面索引对应的元素。tf.nn.embedding_lookup（tensor, id）:tensor就是输入张量，id就是张量对应的索引。

二、代码

import tensorflow as tf
import numpy as np

input_ids = tf.placeholder(dtype=tf.int32, shape=[None])

embedding = tf.Variable(np.identity(5, dtype=np.int32))
input_embedding = tf.nn.embedding_lookup(embedding, input_ids)

sess = tf.InteractiveSession()
sess.run(tf.global_variables_initializer())
print(embedding.eval())
print(sess.run(input_embedding, feed_dict={input_ids:[1, 2, 3, 0, 3, 2, 1]}))