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