單位正規化層
UnitNormalization 類別
tf_keras.layers.UnitNormalization(axis=-1, **kwargs)
單位正規化層。
正規化一批輸入，使批次中的每個輸入的 L2 範數等於 1（跨 axis 中指定的軸）。
範例
>>> data = tf.constant(np.arange(6).reshape(2, 3), dtype=tf.float32)
>>> normalized_data = tf.keras.layers.UnitNormalization()(data)
>>> print(tf.reduce_sum(normalized_data[0, :] ** 2).numpy())
1.0
參數
- axis: 整數或列表/元組。要正規化的軸或多個軸。通常，這是特徵軸或多個軸。剩下的軸通常是批次軸或多個軸。
-1是輸入中的最後一個維度。預設為-1。