何時使用tf.while_loop（）？

Question

這更像是有關何時使用tf.while_loop的一般問題。 例如，如果您有一個可以用固定迭代次數的循環構造的圖，則使用tf.while_loop似乎毫無意義。 記住，只有在不清楚在圖形計算之前必須執行while循環多長時間（即條件取決於必須計算的張量）時，才使用該函數才有意義。 如果我錯了，請指出正確的方向。

Answer 1

tf.while_loop使圖更緊湊，因為它避免了重復。 取決於代碼，可能會帶來一些優勢。 例如，它可以使用較少的資源，使意圖更清晰，並且圖形更易讀且更適合優化。 在最壞的情況下，它會比展開循環要慢一些。

這與普通的while循環與其展開的版本之間的權衡類似：

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

和

x = f(x)
x = f(x)
x = f(x)
x = f(x)