tensorflow::ops::FusedBatchNorm
#include <nn_ops.h>
Нормализация по батчу.
Краткое описание
Обратите внимание, что размер 4D тензоров определяется либо как "NHWC", либо как "NCHW". Размер 1D тензоров соответствует размеру C 4D тензоров.
Аргументы:
- scope: Объект Scope
- x: 4D Tensor для входных данных.
- scale: 1D Tensor для коэффициента масштабирования, для масштабирования нормализованного x.
- offset: 1D Tensor для смещения, для сдвига нормализованного x.
- mean: 1D Tensor для выборочной средней величины. Используется только для вывода; должен быть пустым для обучения.
- variance: 1D Tensor для выборочной дисперсии. Используется только для вывода; должен быть пустым для обучения.
Необязательные атрибуты (см. Attrs):
- epsilon: Малое число с плавающей точкой, добавляемое к дисперсии x.
- data_format: Формат данных для x и y. Либо "NHWC" (по умолчанию), либо "NCHW".
- is_training: Логическое значение, указывающее, что операция предназначена для обучения (по умолчанию) или вывода.
Возвращает:
-
Outputy: 4D Tensor для выходных данных. -
Outputbatch_mean: 1D Tensor для вычисленного среднего значения батча, используемого TensorFlow для вычисления среднего значения. -
Outputbatch_variance: 1D Tensor для вычисленной дисперсии батча, используемой TensorFlow для вычисления дисперсии. -
Outputreserve_space_1: 1D Tensor для вычисленного среднего значения батча, используемого для повторного использования в вычислении градиента. -
Outputreserve_space_2: 1D Tensor для вычисленной дисперсии батча (обратной дисперсии в случае cuDNN), используемой для повторного использования в вычислении градиента.
| Конструкторы и деструкторы | |
|---|---|
FusedBatchNorm(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance) | |
FusedBatchNorm(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNorm::Attrs & attrs) |
| Общедоступные атрибуты | |
|---|---|
batch_mean | |
batch_variance | |
operation | |
reserve_space_1 | |
reserve_space_2 | |
y | |
| Общедоступные статические функции | |
|---|---|
DataFormat(StringPiece x) | |
Epsilon(float x) | |
ExponentialAvgFactor(float x) | |
IsTraining(bool x) | |
| Структуры | |
|---|---|
| tensorflow::ops::FusedBatchNorm::Attrs | Необязательные параметры настройки FusedBatchNorm. |
Общедоступные атрибуты
batch_mean
::tensorflow::Output batch_mean
batch_variance
::tensorflow::Output batch_variance
operation
Operation operation
reserve_space_1
::tensorflow::Output reserve_space_1
reserve_space_2
::tensorflow::Output reserve_space_2
y
::tensorflow::Output y
Общедоступные функции
FusedBatchNorm
FusedBatchNorm( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance )
FusedBatchNorm
FusedBatchNorm( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNorm::Attrs & attrs )
Общедоступные статические функции
DataFormat
Attrs DataFormat( StringPiece x )
Epsilon
Attrs Epsilon( float x )
ExponentialAvgFactor
Attrs ExponentialAvgFactor( float x )
IsTraining
Attrs IsTraining( bool x )
© 2020 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.4/api_docs/cc/class/tensorflow/ops/fused-batch-norm