tensorflow学习笔记2：tf.nn.embedding_lookup

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]]]