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

单位归一化层

[source]

UnitNormalization class

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

单位归一化层.

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

示例:

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

参数: axis: 整数或列表/元组.要归一化的轴或轴.通常,这是特征轴或轴. 被忽略的轴通常是批次轴或轴.-1是输入中的最后一个维度.默认为-1.