tf.nn.embedding_lookup（tensor, id）:
函数作用：
选取一个张量里面索引对应的元素；
函数参数：
tensor：输入的张量；
id：选取张量对应的索引。
复制代码
1
2
3
4
5
6
7a=[[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))
复制代码
1
2
3
4
5
6
7
8out: [[[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内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复