tf.gather（）用法

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Eric_LH/article/details/83794038

tf.gather(等待被取元素的张量，索引)

tf.gather根据索引，从输入张量中依次取元素，构成一个新的张量。

索引的维度可以小于张量的维度。这时，取张量元素时，会把相应的低维当作一个整体取出来。

例如

假设输入张量 [[1,2,3],[4,5,6],[7,8,9]] 是个二维的

如果只给一个一维索引0. 它就把[1,2,3]整体取出：

如果给两个一维索引，0和1，它就形成[[1,2,3],[4,5,6]]