Numpy相当于Tensorflow的embedding_lookup函数

Question

Tensorflow的embedding_lookup函数的NumPy等效代码是什么？

特别是，下面代码块的最后一行的NumPy等价是什么？

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

我不确定embedding_lookup实际上做了什么。

Answer 1

tf.nn.embedding_lookup基本上与tf.gather相似（如上图所示）。它只获得第i个参数片段（“i”是索引元素），并按索引顺序连接切片。