Created by: Colorfu1

我现在得到了一个(B, N, K)大小的索引index，想从(B, N, C)的数据D中取出索引指向的那一行的向量。 请问该怎么使用fluid.layers.gather这个函数进行提取？ 在torch中直接D(index, :)就可以，在tensorflow中使用tf.gather(D, index)就可以。 B的维度一直都是-1。目前尝试过把index进行reshape，变成(B, ) , (B, NK)和(B, 1, NK)的大小，都没有办法进行索引，请问我要怎么做才能正确进行操作。 或者有其他的函数可以这么做吗？ 谢谢