如何以张量作为范围运行循环？(在tensorflow中)

Question

我想要一个for循环,它的迭代次数取决于张量值.例如:

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

但是我收到以下错误:

"TypeError:'Tensor'对象不可迭代"

我该怎么办？

Answer 1

为此,您需要使用tensorflow while loop(tf.while_loop),如下所示:

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

Answer 2

TensorFlow Python API函数的返回值类型，tf.range包括Tensor. ATensor是图中节点的符号句柄，表示计算。eval您可以通过调用a 的方法Tensor或将对象传递给runa 的方法来执行实际计算Session。就您而言，也许您想做的只是简单地迭代numpys range。

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