tf.clip_by_value(t, clip_value_min, clip_value_max,name=None)
tf.clip_by_value(A, min, max):输入一个张量A,把A中的每一个元素的值都压缩在min和max之间。小于min的让它等于min,大于max的元素的值等于max。
import tensorflow as tf import numpy as np A = np.array([[1.2, 0.4, 1, -0.7], [0.3, 1.4, 0.8, -0.5]]) with tf.Session() as sess: print(sess.run(tf.clip_by_value(A, 0, 1)))
结果: