TensorFlowテンソルでリストをインデックスする方法は？

Question

ルックアップテーブルを介してアクセスする必要がある、連結できないオブジェクトのリストを想定します。リストインデックスはテンソルオブジェクトになりますが、これは不可能です。

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 

テンソル付きインデックスリストの方法ですか？回避策ですか？

Answer 1

私はこれが役立つと思う：「numpyの配列にテンソルから戻って変換するにはあなたは、単に変換テンソル上）（.evalを実行することができます」 How can I convert a tensor into a numpy array in TensorFlow?

を

Answer 2

Tensorflowは実際にHashTableをサポートしています。詳細はdocumentationを参照してください。ここで

、何を行う可能性は以下の通りです：

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

それからちょうど鍵が見つからない場合-1はデフォルト値であることを

target = table.lookup(index) 

注意を実行することにより、所望の入力を取得します。テンソルの設定に応じて、key_dtypeとvalue_dtypeをコンストラクタに追加する必要があります。

Answer 3

tf.gatherは、この目的のために設計されています。

tf.gather(list, tf_look_up[index])を実行するだけで、あなたが望むものを手に入れることができます。