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

UnitNormalization 层

[源代码]

UnitNormalization

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

单元归一化层。

对输入批次进行归一化,使批次中的每个输入在指定的 axis 上具有 L2 范数等于 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