def normalize(tensor):
"""
Squeeze the given Tensor into a range between 0 and 1.
:param Tensor tensor: An image tensor.
:return: Tensor
"""
floor = tf.reduce_min(tensor)
ceil = tf.reduce_max(tensor)
return (tensor - floor) / (ceil - floor)
评论列表
文章目录