tf.keras.ops.batch_normalization

Нормализует x по mean и variance.

Псевдонимы

Основные псевдонимы

tf.keras.ops.nn.batch_normalization

tf.keras.ops.batch_normalization(
    x, mean, variance, axis, offset=None, scale=None, epsilon=0.001
)

Этот оператор обычно используется в процессе пакетной нормализации в нейронной сети. Он нормализует входной тензор вдоль указанной оси.

Аргументы
`x`	Входной тензор.
`mean`	Вектор среднего значения той же длины, что и `axis` измерение входного тензора.
`variance`	Вектор дисперсии той же длины, что и `axis` измерение входного тензора.
`axis`	Целое число, ось, которая должна быть нормализована.
`offset`	Вектор смещения той же длины, что и `axis` измерение входного тензора. Если не `None`, `offset` добавляется к нормализованному тензору. По умолчанию `None`.
`scale`	Вектор масштабирования той же длины, что и `axis` измерение входного тензора. Если не `None`, нормализованный тензор умножается на `scale`. По умолчанию `None`.
`epsilon`	Малое число с плавающей точкой, добавляемое к дисперсии, чтобы избежать деления на ноль. По умолчанию 1e-3.

Возвращаемое значение
Нормализованный тензор.

Пример:

x = keras.ops.convert_to_tensor(
    [[0.1, 0.2, 0.3], [0.4, 0.5, 0.6], [0.7, 0.8, 0.9]]
)
keras.ops.batch_normalization(
    x,
    mean=[0.4, 0.5, 0.6],
    variance=[0.67, 0.67, 0.67],
    axis=-1
)
array([[-3.6624e-01, -3.6624e-01, -3.6624e-01],
       [-4.6445e-09,  0.0000e+00, -1.8578e-08],
       [ 3.6624e-01,  3.6624e-01,  3.6624e-01]])

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/batch_normalization