tf.nn.embedding_lookup（tensor, id）:
函数作用：
选取一个张量里面索引对应的元素；
函数参数：
tensor：输入的张量；
id：选取张量对应的索引。
a=[[1,2,3],[4,5,6],[7,8,9]]
a = np.asarray(a)
id = [[0, 2, 1], [1, 1, 2]]
out = tf.nn.embedding_lookup(a, id)
with tf.Session() as sess:
print (sess.run(out))
out:
[[[1 2 3]
[7 8 9]
[4 5 6]]
[[4 5 6]
[4 5 6]
[7 8 9]]]
最后
以上就是陶醉悟空最近收集整理的关于tensorflow学习笔记2：tf.nn.embedding_lookup的全部内容，更多相关tensorflow学习笔记2：tf内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复