tf.gather

tf.gather(params,axis=0,indices )

根据维度和索引值来切片。

对params中维度为axis的值按照索引indices进行切片。