View source on GitHub |
A preprocessing layer which rescales input values to a new range.
tf.keras.layers.Rescaling(
scale, offset=0.0, **kwargs
)
This layer rescales every value of an input (often an image) by multiplying by
scale
and adding offset
.
For instance:
To rescale an input in the
[0, 255]
range to be in the[0, 1]
range, you would passscale=1./255
.To rescale an input in the
[0, 255]
range to be in the[-1, 1]
range, you would passscale=1./127.5, offset=-1
.
The rescaling is applied both during training and inference. Inputs can be of integer or floating point dtype, and by default the layer will output floats.
For an overview and full list of preprocessing layers, see the preprocessing guide.
Input shape | |
---|---|
Arbitrary. |
Output shape | |
---|---|
Same as input. |
Args | |
---|---|
scale
|
Float, the scale to apply to the inputs. |
offset
|
Float, the offset to apply to the inputs. |