Есть ли способ выполнить поиск по словарю на основе тензора строки в Tensorflow?

В простом Python я бы сделал что-то вроде

value = dictionary[key]

, Теперь я хотел бы сделать то же самое во время выполнения Tensorflow, когда у меня естьkey как струнный тензор. Что-то вроде

value_tensor = tf.dict_lookup(string_tensor)

было бы здорово.