tf.nn.embedding_lookup

最新推荐文章于 2021-05-20 18:05:16 发布

梦dancing

最新推荐文章于 2021-05-20 18:05:16 发布

阅读量230

点赞数 1

分类专栏： python语法

本文链接：https://blog.csdn.net/qq_15192373/article/details/90302815

版权

python语法专栏收录该内容

17 篇文章 2 订阅

订阅专栏

用法

tf.nn.embedding_lookup 函数的用法主要是选取一个张量里面索引对应的元素。

参数

tf.nn.embedding_lookup（params, ids）:

params：可以是张量也可以是数组等

id：对应的索引

例子

#c = np.random.random([10, 1])  # 随机生成一个10*1的数组
#b = tf.nn.embedding_lookup(c, [1, 3])#查找数组中的序号为1和3的
p=tf.Variable(tf.random_normal([10,1]))#生成10*1的张量
b = tf.nn.embedding_lookup(p, [1, 3])#查找张量中的序号为1和3的
 
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(b))
    #print(c)
    print(sess.run(p))
    print(p)
    print(type(p))

结果

[[0.15791859]
 [0.6468804 ]]
[[-0.2737084 ]
 [ 0.15791859]
 [-0.01315552]
 [ 0.6468804 ]
 [-1.4090979 ]
 [ 2.1583703 ]
 [ 1.4137447 ]
 [ 0.20688428]
 [-0.32815856]
 [-1.0601649 ]]
<tf.Variable 'Variable:0' shape=(10, 1) dtype=float32_ref>
<class 'tensorflow.python.ops.variables.Variable'>