实现embedding的tf.gather函数

这里，开篇我首先介绍一个tf.gather函数容易踩坑的操作，再详细介绍该函数。

坑

tf.gather函数在GPU上运行时，当出现下标越界的情况时，程序不会报错，而是会返回一个全0的向量作为其embedding向量。在cpu上运行时，出现越界的情况就会直接报错。

tf.gather

tf.gather(
    params,  #下标用来取值的张量
    indices,  #下标
    validate_indices=None,
    name=None,
    axis=0
)

该函数的作用就是根据第二个参数提供的下标，从第一个参数中的张量中取出对应位置的张量。