tf.nn.fused_batch_norm
Нормализация по батчам.
tf.nn.fused_batch_norm(
x, scale, offset, mean=None, variance=None, epsilon=0.001, data_format='NHWC',
is_training=True, name=None
)
См. исходный код: Нормализация по батчам: ускорение обучения глубоких сетей за счет уменьшения сдвига внутренних ковариаций; S. Ioffe, C. Szegedy.
| Аргументы | |
|---|---|
x | Входной Tensor 4-мерного тензора. |
scale | 1-мерный Tensor для масштабирования. |
offset | 1-мерный Tensor для смещения. |
mean | 1-мерный Tensor для среднего значения популяции, используемого для вывода. |
variance | 1-мерный Tensor для дисперсии популяции, используемого для вывода. |
epsilon | Маленькое число с плавающей запятой, добавляемое к дисперсии x. |
data_format | Формат данных для x. "NHWC" (по умолчанию) или "NCHW". |
is_training | Булевое значение, указывающее, используется ли операция для обучения или вывода. |
name | Имя этой операции (необязательно). |
| Возвращаемые значения | |
|---|---|
y | 4-мерный тензор для нормализованного, масштабированного и смещенного x. |
batch_mean | 1-мерный тензор для среднего значения x. |
batch_var | 1-мерный тензор для дисперсии x. |
| Исключения | |
|---|---|
ValueError | Если mean или variance не равны None, когда is_training равен True. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/nn/fused_batch_norm
