python - 如何使用 TensorFlow 张量索引列表？

Question

假设一个列表包含不可连接的对象，需要通过查找表访问。所以列表索引将是一个张量对象，但这是不可能的。

 tf_look_up = tf.constant(np.array([3, 2, 1, 0, 4]))
 index = tf.constant(2)
 list = [0,1,2,3,4]

 target = list[tf_look_up[index]]

这将显示以下错误消息。

 TypeError: list indices must be integers or slices, not Tensor

是否有使用张量索引列表的方法/解决方法？

Answer 1

tf.gather 就是为此目的而设计的。

只需运行 tf.gather(list, tf_look_up[index])，你就会得到你想要的。