重新縮放層

[原始碼]

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: 浮點數,要應用於輸入的偏移量。