何时使用tf.while_loop()?

时间:2017-12-03 09:29:30

标签: python tensorflow

这更像是关于何时使用tf.while_loop的一般性问题。例如,如果你有一个可以用固定迭代量的循环构造的图形,那么使用tf.while_loop似乎没有意义。记住,当不清楚在图形计算之前必须执行多长时间时,即当条件依赖于必须计算的张量时,使用该函数似乎才有意义。如果我错了,请指出正确的方向。

1 个答案:

答案 0 :(得分:0)

UI使图形更加紧凑,因为它避免了重复。取决于代码,可能会带来一些优势。例如,它可以使用较少的资源,使意图更清晰,并且图形更易读且更适合优化。在最坏的情况下,它会比展开循环慢一些。

这与普通的while循环及其展开版本之间的权衡相似:

tf.while_loop

i = 0
while i < 4:
    x = f(x)
    i += 1