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

UnitNormalization 层

[源代码]

UnitNormalization

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

单位归一化层。

对输入批次进行归一化,使其在批次中的每个输入沿着axis指定的轴上具有 L2 范数 1。

示例

>>> data = tf.constant(np.arange(6).reshape(2, 3), dtype=tf.float32)
>>> normalized_data = tf.keras.layers.UnitNormalization()(data)
>>> print(tf.reduce_sum(normalized_data[0, :] ** 2).numpy())
1.0

参数

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