tensorflow API: tf.gather

import tensorflow as tf

temp = tf.range(0,10)*10 + tf.constant(1,shape=[10])
temp2 = tf.gather(temp,[1,5,9])

with tf.Session() as sess:

    print(sess.run(temp))
    print(sess.run(temp2))

Output
[ 1 11 21 31 41 51 61 71 81 91]
[11 51 91]

Reference: https://blog.csdn.net/guotong1988/article/details/53172882