tensorflow::ops::FusedBatchNormGrad
#include <nn_ops.h>Градиент для пакетной нормализации.
Краткое содержание
Обратите внимание, что размер 4D-тензоров определяется либо "NHWC", либо "NCHW". Размер 1D-тензоров соответствует размерности C 4D-тензоров.
Аргументы:
- scope: Объект Scope
- y_backprop: Четырехмерный тензор для градиента относительно y.
- x: Четырехмерный тензор для входных данных.
- scale: Одномерный тензор для масштабирования, используемый для масштабирования нормализованного значения x.
- reserve_space_1: Если is_training равно True, то для вычисления градиента используется одномерный тензор для вычисленного среднего значения пакета. Если is_training равно False, то для вычисления градиента первого и второго порядка используется одномерный тензор для среднего значения популяции.
- reserve_space_2: Если is_training равно True, то используется одномерный тензор для вычисленной дисперсии пакета (обратная дисперсия в случае cuDNN) для повторного использования при вычислении градиента. Если is_training равно False, то используется одномерный тензор для дисперсии популяции для повторного использования как при вычислении градиента первого, так и второго порядка.
Дополнительные атрибуты (см. Attrs ):
- Эпсилон: небольшое число с плавающей запятой, добавляемое к дисперсии x.
- data_format: Формат данных для y_backprop, x, x_backprop. Либо "NHWC" (по умолчанию), либо "NCHW".
- is_training: Логическое значение, указывающее, предназначена ли операция для обучения (по умолчанию) или для вывода результатов.
Возвращает:
-
Outputx_backprop: 4D- тензор градиента относительно x. -
Outputпараметр scale_backprop: одномерный тензор градиента относительно масштаба. -
Outputoffset_backprop: Одномерный тензор градиента относительно смещения. -
Outputreserve_space_3: Неиспользуемый заполнитель для сопоставления среднего значения входных данных в FusedBatchNorm . -
Outputreserve_space_4: Неиспользуемый заполнитель для соответствия входным данным дисперсии в FusedBatchNorm .
Конструкторы и Деструкторы | |
|---|---|
FusedBatchNormGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2) | |
FusedBatchNormGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2, const FusedBatchNormGrad::Attrs & attrs) |
Общедоступные атрибуты | |
|---|---|
offset_backprop | |
operation | |
reserve_space_3 | |
reserve_space_4 | |
scale_backprop | |
x_backprop | |
Открытые статические функции | |
|---|---|
DataFormat (StringPiece x) | |
Epsilon (float x) | |
IsTraining (bool x) | |
Структуры | |
|---|---|
| tensorflow::ops::FusedBatchNormGrad::Attrs | Дополнительные параметры установки атрибутов для FusedBatchNormGrad . |
Общедоступные атрибуты
offset_backprop
::tensorflow::Output offset_backprop
операция
Operation operation
резервное_пространство_3
::tensorflow::Output reserve_space_3
резервное_пространство_4
::tensorflow::Output reserve_space_4
scale_backprop
::tensorflow::Output scale_backprop
x_backprop
::tensorflow::Output x_backprop
Общественные функции
FusedBatchNormGrad
FusedBatchNormGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input y_backprop, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input reserve_space_1, ::tensorflow::Input reserve_space_2 )
FusedBatchNormGrad
FusedBatchNormGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input y_backprop, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input reserve_space_1, ::tensorflow::Input reserve_space_2, const FusedBatchNormGrad::Attrs & attrs )
Открытые статические функции
Формат данных
Attrs DataFormat( StringPiece x )
Эпсилон
Attrs Epsilon( float x )
IsTraining
Attrs IsTraining( bool x )