重缩放层

[源]

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

重缩放操作在训练和推理期间都会应用。输入可以是整数或浮点型 dtype,默认情况下,层将输出浮点数。

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

输入形状

任意。

输出形状

与输入相同。

参数

  • scale: 浮点数,应用于输入的缩放比例。
  • offset: 浮点数,应用于输入的偏移量。