我有一个TensorFlow数组，我想使用一个字典将其中的每个元素转换为另一个元素。
这是我的数组：
elems = tf.convert_to_tensor(np.array([1, 2, 3, 4, 5, 6]))这是字典：
d = {1:1,2:5,3:7,4:5,5:8,6:2}转换后，结果数组应为
tf.convert_to_tensor(np.array([1, 5, 7, 5, 8, 2]))为了实现这一点，我尝试使用tf.map_fn，如下所示：
当我运行上述代码时，得到以下错误：
squares = tf.map_fn(lambda x: d[x], elems) KeyError: <tf.Tensor 'map/while/TensorArrayReadV3:0' shape=() dtype=int64>
正确的做法是什么？我基本上是按照这里的使用方法进行操作的。
P.S. 我的数组实际上是3D的，我只是使用1D作为示例，因为代码在这种情况下也会失败。
回答：
你应该使用tensorflow.contrib.lookup.HashTable：
