tensorflow::ops::FusedBatchNormV2
#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), используемой при вычислении градиента.
| Конструкторы и деструкторы | |
|---|---|
FusedBatchNormV2(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance) | |
FusedBatchNormV2(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV2::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::FusedBatchNormV2::Attrs | Дополнительные параметры для FusedBatchNormV2. |
Публичные атрибуты
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
Публичные функции
FusedBatchNormV2
FusedBatchNormV2( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance )
FusedBatchNormV2
FusedBatchNormV2( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV2::Attrs & attrs )
Публичные статические функции
DataFormat
Attrs DataFormat( StringPiece x )
Epsilon
Attrs Epsilon( float x )
ExponentialAvgFactor
Attrs ExponentialAvgFactor( float x )
IsTraining
Attrs IsTraining( bool x )
© 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/cc/class/tensorflow/ops/fused-batch-norm-v2