1. 程式人生 > >TensorFlow函式之tf.clip_by_value()

TensorFlow函式之tf.clip_by_value()

tf.clip_by_value()函式可以將一個張量中的數值限制在一個範圍之內。

下邊通過簡單的例子說明次函式的用法:

import tensorflow as tf
v = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
print(tf.clip_by_value(v, 2.5, 4.5).eval())

輸出為:

[[2.5 2.5 3. ]
 [4.  4.5 4.5]]

輸出結果分析:小於2.5的數被換成了2.5,大於4.5的數被換成了4.5。

使用tf.clip_by_value()可以通過將數值限制在某一範圍之內保證計算的正確性,如在計算log運算時,可以通過限制使得不會出現log0這樣的錯誤。

 例子中出現了eval()函式,有對eval()函式不瞭解的,可以參考本部落格對eval()函式的介紹