深度理解tf.gather和tf.gather_nd的用法

tf.gather和tf.gather_nd都是从tensor中取出index标注的部分，不同之处在于，gather一般只使用一个index来标注，而gather_nd可以使用多个index。
例如：


如何直观理解gather_nd的indices呢？
在上例中，直观的理解就是，取出params中位于[0,0]和[1,1]处的tensor，放入index中对应的位置。
换句话说，除去tensor维之外，返回值的形状和indices相同，值由indices标注。
如果理解了这一点，就可以用gather_nd实现gather：

来源：CSDN

作者：iSikai

链接：https://blog.csdn.net/oksupersonic/article/details/104559821