tf.clip_by_value() 用法

weixin_41874599發表於2018-11-15

tf.clip_by_value(v,a,b)
功能:可以將一個張量中的數值限制在一個範圍之內。(可以避免一些運算錯誤)
引數:(1)v:input資料(2)a、b是對資料的限制。
當v小於a時,輸出a;
當v大於a小於b時,輸出原值;
當v大於b時,輸出b;
例子:

import tensorflow as tf

v=tf.constant([[1.0,2.0,3.0],[4.0,5.0,6.0]])

sess=tf.Session()

with sess.as_default():

    print(tf.clip_by_value(v,2.5,4.5).eval())
 

  結果:
[[ 2.5 2.5 3. ]
[ 4. 4.5 4.5]]
-