tf.clip_by_value()函数的功能是将一个张量中的数值限制在一个范围之内。
这样可以避免一些运算错误:例如在进行log运算时,不会出现log0这样的错误
实际使用过程中,第一个参数是输入的数据,后两个参数分别是限制的最小值和最大值。
代码示例:
import tensorflow as tf
a = tf.constant([1, 2, 3, 4, 5])
sess = tf.Session()
a = tf.clip_by_value(a, 2, 4)
with tf.Session() as sess:
print(sess.run(a))
输出为: