这更像是关于何时使用tf.while_loop的一般性问题。例如,如果你有一个可以用固定迭代量的循环构造的图形,那么使用tf.while_loop似乎没有意义。记住,当不清楚在图形计算之前必须执行多长时间时,即当条件依赖于必须计算的张量时,使用该函数似乎才有意义。如果我错了,请指出正确的方向。
答案 0 :(得分:0)
UI
使图形更加紧凑,因为它避免了重复。取决于代码,可能会带来一些优势。例如,它可以使用较少的资源,使意图更清晰,并且图形更易读且更适合优化。在最坏的情况下,它会比展开循环慢一些。
这与普通的while循环及其展开版本之间的权衡相似:
tf.while_loop
和
i = 0
while i < 4:
x = f(x)
i += 1