Tensorflow embedding_lookup函数的Numpy等价

Question

什么是与Tensorflow的embedding_lookup函数等价的embedding_lookup代码？

特别是，以下代码块的最后一行的NumPy等效值是什么？

words = tf.placeholder(tf.int64, name='words')
...
embedding = tf.nn.embedding_lookup(embedding_params, words[:, i])

我不太清楚embedding_lookup到底做了什么。

Answer 1

​

​

查找的工作方式基本类似于tf.gather(在上面的图片中)。它只得到第一层的参数(“i”是索引的元素)，并按索引的顺序连接切片。