如何使用 TensorFlow 张量索引列表？答案

【问题标题】：How to index a list with a TensorFlow tensor?如何使用 TensorFlow 张量索引列表？
【发布时间】：2023-03-15 22:35:01
【问题描述】：

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

 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

是用张量索引列表的方法/解决方法吗？

【问题讨论】：

【解决方案1】：

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

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

【讨论】：

【解决方案2】：

Tensorflow 实际上支持HashTable。有关详细信息，请参阅documentation。

在这里，您可以执行以下操作：

table = tf.contrib.lookup.HashTable(
    tf.contrib.lookup.KeyValueTensorInitializer(tf_look_up, list), -1)

然后通过运行获得所需的输入

target = table.lookup(index)

请注意，如果找不到密钥，-1 是默认值。根据张量的配置，您可能必须将key_dtype 和value_dtype 添加到构造函数中。

【讨论】：