用法：

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

• cond 是一個(gè)函數(shù)，負(fù)責(zé)判斷繼續(xù)執(zhí)行循環(huán)的條件。
• loop_body 是每個(gè)循環(huán)體內(nèi)執(zhí)行的操作，負(fù)責(zé)對(duì)循環(huán)狀態(tài)迸行更新。
• init_state 為循環(huán)的起始狀態(tài)，它可以包含多個(gè) Tensor 或者 TensorArray 。
• 返回的結(jié)果是循環(huán)結(jié)束時(shí)的循環(huán)狀態(tài)。

如果用偽代碼來表示運(yùn)行邏輯的話，那 tf.while_loop 的功能與下面的代碼相當(dāng) ：

def while_loop(cond, loop_body, init_state): state = init_state while(cond(state)) : ＃ 使用cond函數(shù)判斷是否達(dá)到循環(huán)結(jié)束條件。state = loop_body(state) ＃ 使用loop_body函數(shù)對(duì)state進(jìn)行更新。return state

?

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。