どのようにテンソルを範囲としてループを実行できますか？ （テンソルフローで）

Question

繰り返しの回数がテンソルの値に依存するforループが必要です。たとえば：

for i in tf.range(input_placeholder[1,1]): 
    # do something 

私は次のエラーを取得するしかし：

"はTypeError： 'テンソル' オブジェクトが反復可能ではありません"

私は何をすべきか？

Answer 1

tf.rangeを含むTensorFlow Python API関数の戻り値の型は、Tensorです。 Tensorは、計算を表すグラフのノードへの記号ハンドルです。実際の計算は、Tensorのevalメソッドを呼び出すか、またはのrunメソッドにオブジェクトを渡すことで行います。あなたの場合、おそらくあなたが意図したのはnumpyのrangeを繰り返し処理するだけでした。あなたがループ（tf.while_loop）ながらtensorflowを使用する必要があります。これを行うために

for in in np.range(...): 
    # do something 

Answer 2

次のように：

i = tf.constant(0) 
while_condition = lambda i: tf.less(i, input_placeholder[1, 1]) 
def body(i): 
    # do something here which you want to do in your loop 
    # increment i 
    return [tf.add(i, 1)] 

# do the loop: 
r = tf.while_loop(while_condition, body, [i])