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

Question

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

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

但是我得到以下错误:

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

我该怎么办？

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