Come posso eseguire un ciclo con un tensore come intervallo? (in tensorflow)

Question

Voglio avere un ciclo for che il numero delle sue iterazioni dipenda da un valore del tensore. Per esempio:

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

Tuttavia ottengo il seguente errore:

"TypeError: oggetto 'Tensor' non è iterabile"

Cosa devo fare?

Answer 1

Il tipo del valore restituito delle funzioni API Python TensorFlow, incluso tf.range è un Tensor. A Tensor è un handle simbolico per il nodo in un grafico che rappresenta il calcolo. Si esegue il calcolo effettivo chiamando il metodo eval su un Tensor o passando l'oggetto al metodo run di un Session. Nel tuo caso, forse quello che avevi intenzione di fare era semplicemente scorrere lo range dello numpy.

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

Answer 2

Per fare questo è necessario utilizzare il tensorflow ciclo while (tf.while_loop) come segue:

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