是否有numpy的方法与tensorflow的embedding_lookup（）相同？

Question

我想通过indice

从矩阵中查找

如果矩阵看起来像这样

  

1 2 3 4

     

5 6 7 8

     

9 2 1 3

并且有像

这样的指标

  

[1,1,2,0]

所以，我想要做的就是获取

  

5 6 7 8

     

5 6 7 8

     

9 2 1 3

     

1 2 3 4

我可以使用tensorflow的embedding_lookup()或gather()

来完成

但是由于我的搜索能力差，我在找到这种操作的方法时遇到了麻烦。

有这种操作的方法吗？