tf.clip_by_value()

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]]

猜你喜欢

转载自blog.csdn.net/qq_30638831/article/details/80670773
今日推荐