Spec-Zone.ru › TensorFlow 2.9

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/versions/r2.9/api_docs/python/tf/compat/v1/nn/fused_batch_norm

Spec-Zone.ru

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