while_loop是這樣的:i = tf.constant(0)c = lambda i: tf.less(i, 10)b = lambda i: tf.add(i, 1)r = tf.while_loop(c, b, [i])i用作增量變量。因此,i是可變的。為什么我們定義i為常數?為什么不i = tf.Variable(0, tf.int32)
2 回答

LEATH
TA貢獻1936條經驗 獲得超7個贊
i
用作增量變量。因此,i
是可變的。
不完全正確-如果是,i
會不會是一個常量。
tf.add(i, 1)
不會改變i
,它需要張量i
并通過添加1來創建新的張量。(在實踐中,張量流可能會為結果張量重用相同的內存分配,但這是與的邏輯無關的優化tf.where
)。
您可能會感到困惑,因為i
在lambda中使用了相同的名稱,但是所有這些張量都是與每次迭代操作的輸出相對應的不同張量。
因此,i
您聲明的實際上是迭代的第一個值(循環中的所有其他值,不僅是計數器,都是相同的)。它確實是一個常數0,因此將其聲明為常量是有意義的。將初始值0聲明為a會很尷尬Variable
。
添加回答
舉報
0/150
提交
取消