概述
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.nn.embedding_lookup所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复