tf.keras.layers.Rescaling

View source on GitHub

A preprocessing layer which rescales input values to a new range.

Inherits From: Layer, Module

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:

1. To rescale an input in the [0, 255] range to be in the [0, 1] range, you would pass scale=1./255.

2. To rescale an input in the [0, 255] range to be in the [-1, 1] range, you would pass scale=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.