2016-02-10 18 views
12

Tensorflow'da bir String tensörüne dayalı sözlük araması gerçekleştirmenin herhangi bir yolu var mı?Tensorflow Dictionary with String tensör

düz Python, ben

value = dictionary[key] 

gibi bir şey yapardım. Şimdi aynı şeyi Tensorflow çalışma zamanında yapmak istiyorum, key benim bir String tensörü olarak.

gibi bir şey hoş olurdu.

cevap

-4

TensorFlow, tensör dışındaki veri yapıları için destek içermeyen bir veri akışı dilidir. Harita veya sözlük türü yok. Bununla birlikte, ihtiyacınız olan şeye bağlı olarak, Python sarıcısını kullandığınızda, Python'da yürüten ve TensorFlow grafik uygulamasıyla etkileşime geçmek için sürücü işleminde bir sözlüğü korumak mümkündür. Örneğin, bir oturumda bir TensorFlow grafiğinin bir adımını yürütebilir, Python sürücüsüne bir dize değeri döndürür, bunu sürücüdeki bir sözlükte anahtar olarak kullanabilir ve istenen bir sonraki hesaplamayı belirlemek için alınan değeri kullanabilirsiniz. oturumdan Bu sözlük aramalarının hızı, performans açısından kritik ise, bu muhtemelen iyi bir çözüm değildir.

+0

https://www.tensorflow.org/api_docs/python/tf/ contrib/lookup/HashTable – Julius

14

Sen tensorflow.contrib.lookup yararlı bulabilirsiniz: Özellikle https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lookup/lookup_ops.py

https://www.tensorflow.org/api_docs/python/tf/contrib/lookup/HashTable

, yapabileceğiniz:

table = tf.contrib.lookup.HashTable(
    tf.contrib.lookup.KeyValueTensorInitializer(keys, values), -1 
) 
out = table.lookup(input_tensor) 
table.init.run() 
print out.eval() 
+2

https://www.tensorflow.org/api_docs/python/tf/contrib/lookup/HashTable – Julius

İlgili konular