tf.while_loop,tf.cast 轉換數據類型。

tf.while_loop 可以這樣理解

loop = []
while cond(loop):
    loop = body(loop)

即loop參數先傳入cond 判斷條件是否成立,成立之後,把 loop參數傳入body 執行操作, 然後返回 操作後的 loop 參數,即loop參數已被更新,再把更新後的參數傳入cond, 依次循環,直到不滿足條件。

tf.cast 轉換數據類型。

a = tf.constant([0, 2, 0, 4, 2, 2], dtype='int32')
print(a)
# <tf.Tensor 'Const_1:0' shape=(6,) dtype=int32>

b = tf.cast(a, 'float32')
print(b)
# <tf.Tensor 'Cast:0' shape=(6,) dtype=float32>

tf.greater

首先張量x和張量y的尺寸要相同,輸出的tf.greater(x, y)也是一個和x,y尺寸相同的張量。如果x的某個元素比y中對應位置的元素大,則tf.greater(x, y)對應位置返回True,否則返回False。

tf.less
tensorflow的內置函數,寫x,y都是一個tensor,返回值是一個bool型的tensor.[逐元素返回是否x<y]

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章