Rescaling 层

[源代码]

Rescaling

tf_keras.layers.Rescaling(scale, offset=0.0, **kwargs)

一个预处理层,用于将输入值重新缩放到新的范围。

此层通过乘以 scale 并加上 offset 来重新缩放输入的每个值(通常是图像)。

例如

  1. 要将范围在 [0, 255] 的输入重新缩放到范围 [0, 1],可以传入 scale=1./255

  2. 要将范围在 [0, 255] 的输入重新缩放到范围 [-1, 1],可以传入 scale=1./127.5, offset=-1

重新缩放在训练和推理期间都适用。输入可以是整数或浮点数类型,默认情况下,该层将输出浮点数。

有关预处理层的概述和完整列表,请参阅预处理 指南

输入形状

任意。

输出形状

与输入相同。

参数

  • scale: Float,应用于输入的比例因子。
  • offset: Float,应用于输入的偏移量。