Spec-Zone.ru › TensorFlow

tf.compat.v1.nn.fused_batch_norm

Нормализация по батчу.

tf.compat.v1.nn.fused_batch_norm(
    x,
    scale,
    offset,
    mean=None,
    variance=None,
    epsilon=0.001,
    data_format='NHWC',
    is_training=True,
    name=None,
    exponential_avg_factor=1.0
)

См. исходный код: Нормализация по батчу: ускорение обучения глубоких сетей за счёт снижения смещения внутренних ковариатов; С. Иоффе, К. Зегеди.

Аргументы
x Входной тензор Tensor 4 или 5 измерений.
scale Массив Tensor 1 измерения для масштабирования.
offset Массив Tensor 1 измерения для смещения.
mean Массив Tensor 1 измерения для выборочного среднего. Форма и смысл этого аргумента зависят от значения is_training и exponential_avg_factor следующим образом: is_trainingFalse (вывод): Среднее значение должно быть Tensor такой же формы, как scale, содержащим оценённое выборочное среднее, вычисленное во время обучения. is_trainingTrue и exponential_avg_factor == 1.0: Среднее значение должно быть None. is_trainingTrue и exponential_avg_factor != 1.0: Среднее значение должно быть Tensor такой же формы, как scale, содержащим экспоненциальное скользящее среднее.
variance Массив Tensor 1 измерения для выборочной дисперсии. Форма и смысл этого аргумента зависят от значения is_training и exponential_avg_factor следующим образом: is_trainingFalse (вывод): Дисперсия должна быть Tensor такой же формы, как scale, содержащим оценённую выборочную дисперсию, вычисленную во время обучения. is_training==True и exponential_avg_factor == 1.0: Дисперсия должна быть None. is_training==True и exponential_avg_factor != 1.0: Дисперсия должна быть Tensor такой же формы, как scale, содержащим экспоненциальную скользящую дисперсию.
epsilon Малое число с плавающей точкой, добавляемое к дисперсии x.
data_format Формат данных x. Поддерживаются "NHWC" (по умолчанию) или "NCHW" для 4D тензоров и "NDHWC" или "NCDHW" для 5D тензоров.
is_training Булево значение, указывающее, используется ли операция для обучения или вывода.
name Имя этой операции (необязательно).
exponential_avg_factor Число с плавающей точкой (обычно между 0 и 1), используемое для управления затуханием экспоненциального скользящего среднего выборочного среднего и дисперсии. Если установлено в 1.0, возвращается текущее выборочное среднее.
Возвращаемые значения
y 4D или 5D тензор для нормализованного, масштабированного и смещённого x.
running_mean 1D тензор для экспоненциального скользящего среднего x. Значение результата равно (1 - exponential_avg_factor) * mean + exponential_avg_factor * batch_mean), где batch_mean — среднее значение текущего батча в x.
running_var 1D тензор для экспоненциального скользящего отклонения. Значение результата равно (1 - exponential_avg_factor) * variance + exponential_avg_factor * batch_variance), где batch_variance — дисперсия текущего батча в x.
Ссылки
Нормализация по батчу - ускорение обучения глубоких сетей за счёт снижения смещения внутренних ковариатов: Ioffe et al., 2015 (pdf)

© 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/compat/v1/nn/fused_batch_norm

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API