tensorflow 中tf.gather(params, indices, validate_indices=None, name=None) 函数讲解

tf.gather

tf.gather(params, indices, validate_indices=None, name=None, axis=0)

params 表示你输入的张量，indices表示你想要params张量中切片的维度，所以这个函数就是挑选出params中indices对应的数。

举例子

x = tf.constant(np.arange(8).reshape((2,2,2)))y = tf.gather(x,[0])sess  = tf.Session()print(sess.run(y))print('---------')print(sess.run(x))

[[[0 1]  [2 3]]]---------[[[0 1]  [2 3]] [[4 5]  [6 7]]]

就相当把x矩阵的第一个切片取出来了
想要其他轴切片，可以设置axis这个参数