Keras 3 API 文档 / 层 API / 归一化层 / UnitNormalization 层

UnitNormalization 层

[源码]

UnitNormalization

keras.layers.UnitNormalization(axis=-1, **kwargs)

Unit 归一化层。

对一批输入进行归一化,使批次中每个输入的 L2 范数等于 1(在 axis 中指定的轴上)。

示例

>>> data = np.arange(6).reshape(2, 3)
>>> normalized_data = keras.layers.UnitNormalization()(data)
>>> np.sum(normalized_data[0, :] ** 2)
1.0

参数

  • axis: 整数或列表/元组。要进行归一化的轴。通常,这是特征轴。剩余的轴通常是批次轴。-1 是输入中的最后一个维度。默认为 -1