Question

0

我想要一个for循环，其迭代次数取决于张量值。例如：

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

但是我收到以下错误：

“ TypeError：'Tensor'对象不可迭代”

我该怎么办？

参考资料：

Stack Overflow

Answer 1

为此，您将需要使用tensorflow while循环（ 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])