Keras 2 API 文档 / 层 API / 归一化层 / 单元归一化层 (UnitNormalization layer)

单元归一化层 (UnitNormalization layer)

[源码]

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`。