Keras 3 API 文档 / 层 API / 归一化层 / 单元归一化层

单元归一化层

[源代码]

UnitNormalization

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

单元归一化层。

将一批输入进行归一化,使得批次中的每个输入在 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