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/api_docs/cc/class/tensorflow/ops/fused-batch-norm-v2