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
参数
-1 表示输入的最后一个维度。默认为 -1。