tensorflow::ops::FusedBatchNormV3
#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), повторно используемым при вычислении градиента. -
Outputreserve_space_3: 1D-Tensor с некоторыми промежуточными результатами, повторно используемыми при вычислении градиента для повышения эффективности.
| Конструкторы и деструкторы | |
|---|---|
FusedBatchNormV3(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance) | |
FusedBatchNormV3(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV3::Attrs & attrs) |
| Открытые атрибуты | |
|---|---|
batch_mean | |
batch_variance | |
operation | |
reserve_space_1 | |
reserve_space_2 | |
reserve_space_3 | |
y | |
| Открытые статические функции | |
|---|---|
DataFormat(StringPiece x) | |
Epsilon(float x) | |
ExponentialAvgFactor(float x) | |
IsTraining(bool x) | |
| Структуры | |
|---|---|
| tensorflow::ops::FusedBatchNormV3::Attrs | Методы настройки необязательных атрибутов для FusedBatchNormV3. |
Открытые атрибуты
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
reserve_space_3
::tensorflow::Output reserve_space_3
y
::tensorflow::Output y
Открытые функции
FusedBatchNormV3
FusedBatchNormV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance )
FusedBatchNormV3
FusedBatchNormV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV3::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/api_docs/cc/class/tensorflow/ops/fused-batch-norm-v3