如何在張量流中調整張量？

Question

例如，我有一個張量形狀[128,128,3]，它的範圍是隨機的，那麼我想將這個張量中的所有數字都縮放到[0,255]，我應該使用張量流中的什麼函數來做到這一點？ 謝謝

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 

你只需要處理邊緣情況，當所有的值都相等。