用法：

final_state = tf.while_loop(cond, loop_body, init_state)

• cond 是一個函數，負責判斷繼續執行循環的條件。
• loop_body 是每個循環體內執行的操作，負責對循環狀態迸行更新。
• init_state 為循環的起始狀態，它可以包含多個 Tensor 或者 TensorArray 。
• 返回的結果是循環結束時的循環狀態。

如果用偽代碼來表示運行邏輯的話，那 tf.while_loop 的功能與下面的代碼相當 ：

def while_loop(cond, loop_body, init_state): state = init_state while(cond(state)) : ＃ 使用cond函數判斷是否達到循環結束條件。state = loop_body(state) ＃ 使用loop_body函數對state進行更新。return state

?

總結

以上是生活随笔為你收集整理的tf.while_loop的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。