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

UnitNormalization 层

[源]

UnitNormalization

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

Unit 归一化层。

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

示例

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

参数

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