您可能会发现tensorflow.contrib.lookup
有用: https : //github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lookup/lookup_ops.py
https://www.tensorflow.org/api_docs/python/tf/contrib/lookup/HashTable
特别是,您可以执行以下操作:
table = tf.contrib.lookup.HashTable(
tf.contrib.lookup.KeyValueTensorInitializer(keys, values), -1
)
out = table.lookup(input_tensor)
table.init.run()
print out.eval()
0
有没有办法在Tensorflow中基于字符串张量执行字典查找?
在普通的Python中,我会做类似的事情
。现在,当我将我的
key
作为String张量时,我想在Tensorflow运行时做同样的事情。就像是会好的。