如何在张量流中缩放张量？

Question

例如，我的张量形状是[128,128,3]并且它的范围是随机的，那么我想将此张量中的所有num缩放到[0,255]，我应该使用tensorflow中的哪个函数来做到了吗？ 感谢

Answer 1

您可以自己进行缩放。

// x is your tensor
current_min = tf.reduce_min(x)
current_max = tf.reduce_max(x)
target_min = 0
target_max = 255

// scale to [0; 1]
x = (x - current_min) / (current_max - current_min)

// scale to [target_min; target_max]
x = x * (target_max - target_min) + target_min

当所有值相等时，您只需处理边缘情况。