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) | |
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 )
IsTraining
Attrs IsTraining( bool x )
© 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/cc/class/tensorflow/ops/fused-batch-norm-v2