Jak mogę uruchomić pętlę z tensorem jako jej zasięg? (w tensorflow)

Question

Chcę mieć pętlę for, której liczba iteracji zależy od wartości tensora. Na przykład:

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

Jednak pojawia się następujący błąd:

"TypeError: obiekt 'tensor' nie jest iterable"

Co należy zrobić?

Answer 1

Typ zwracanej wartości funkcji API TensorFlow Python, w tym tf.range, to Tensor. A Tensor to symboliczny uchwyt do węzła na wykresie reprezentującym obliczenia. Wykonujesz rzeczywiste obliczenia, wywołując metodę eval na Tensor lub przekazując obiekt do metody run z Session. W twoim przypadku być może to, co masz zamiar zrobić, to po prostu powtórzyć numpy 's range.

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

Answer 2

W tym celu trzeba będzie użyć tensorflow pętli while (tf.while_loop) w następujący sposób zrobić:

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])