Method call

tfjs-layers/src/layers/preprocessing/image_preprocessing.ts:55–63

(inputs: Tensor|Tensor[], kwargs: Kwargs)

  }

  override call(inputs: Tensor|Tensor[], kwargs: Kwargs): Tensor[]|Tensor {
    return tidy(() => {
      inputs = getExactlyOneTensor(inputs);
      if(inputs.dtype !== 'float32') {
          inputs = K.cast(inputs, 'float32');
      }
      return add(mul(inputs, this.scale), this.offset);
    });
  }
}

serialization.registerClass(Rescaling);