参考tf.clip_by_value( ) - 云+社区 - 腾讯云
tf.clip_by_value的用法:
tf.clip_by_value(A, min, max):输入一个张量A,把A中的每一个元素的值都压缩在min和max之间。小于min的让它等于min,大于max的元素的值等于max。
tf.clip_by_value(
t,
clip_value_min,
clip_value_max,
name=None
)
给定一个张量t,这个操作返回一个与t相同类型和形状的张量,其值被剪切到clip_value_min和clip_value_max。任何小于clip_value_min的值都被设置为clip_value_min。任何大于clip_value_max的值都被设置为clip_value_max。
参数:
- t:张量或转位切片。
- clip_value_min:一个0-D(标量)张量,或者一个与t形状相同的张量。
- clip_value_max:一个0-D(标量)张量,或者一个与t形状相同的张量。
- name:操作的名称(可选)。
返回值:
- 剪切张量或索引切片。
可能产生的异常:
ValueError
: If the clip tensors would trigger array broadcasting that would make the returned tensor larger than the input.TypeError
: If dtype of the input isint32
and dtype of theclip_value_min' or
clip_value_maxis
float32`
例:
import tensorflow as tf;
import numpy as np;
A = np.array([[1,1,2,4], [3,4,8,5]])
with tf.Session() as sess:
print sess.run(tf.clip_by_value(A, 2, 5))
运行结果如下: