TensorFlow:tf.clip_by_value实现值域控制

tf.clip_by_value(A, min, max)函数:输入一个张量A,把A中的每一个元素的值都压缩在min和max之间。小于min的让它等于min,大于max的元素值等于max。

代码实现:

import tensorflow as tf

A = tf.constant([1,2,288,20,100])
value = tf.clip_by_value(A, 10, 255)

with tf.Session() as sess:
    print(sess.run(value))

result:[ 10 10 255 20 100]

你可能感兴趣的:(Tensorflow的学习)