tf.keras.layers.UnitNormalization

Слой единичной нормализации.

Наследуется от: Layer, Module

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