TensorFlow：使用张量索引另一个张量

TensorFlow：使用张量索引另一个张量

幸运的是，TensorFlow支持您所要求的确切情况

tf.gather()

：

result = x_t * tf.gather(e_t, x_t)with tf.Session() as sess:    print sess.run(result)  # ==> 'array([1, 0, 3, 3, 0, 5, 0, 7, 1, 3])'

该

tf.gather()

运算是小于强大NumPy的先进索引：它仅支持其零维提取张量的全片。有人要求支持更一般的索引编制，并且此GitHub问题中对此进行了跟踪。