如何运行以张量为范围的循环? (在张量流中)
deep-learning
neural-network
python
tensorflow
5
0

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

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

但是我收到以下错误:

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

我该怎么办?

参考资料:
Stack Overflow
收藏
评论
共 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])
收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号