тензорный поток:: опс:: FusedBatchНормГрад
#include <nn_ops.h>
Градиент для пакетной нормализации.
Краткое содержание
Обратите внимание, что размер 4D-тензоров определяется либо «NHWC», либо «NCHW». Размер 1D-тензоров соответствует размеру C 4D-тензоров.
Аргументы:
- область: объект области.
- y_backprop: 4D- тензор для градиента по y.
- x: 4D- тензор для входных данных.
- масштаб: 1D- тензор для коэффициента масштабирования, чтобы масштабировать нормализованный x.
- Reserve_space_1: Когда is_training имеет значение True, одномерный тензор для вычисляемого пакета означает повторное использование при вычислении градиента. Когда is_training имеет значение False, одномерный тензор для генеральной совокупности будет повторно использоваться при вычислении градиента как 1-го, так и 2-го порядка.
- резерв_пространство_2: когда is_training имеет значение True, 1D- тензор для вычисленной пакетной дисперсии (инвертированной дисперсии в случае cuDNN), который будет повторно использоваться при вычислении градиента. Когда is_training имеет значение False, 1D- тензор для дисперсии генеральной совокупности будет повторно использоваться при вычислении градиента как 1-го, так и 2-го порядка.
Необязательные атрибуты (см. Attrs
):
- эпсилон: небольшое число с плавающей запятой, добавляемое к дисперсии x.
- data_format: формат данных для y_backprop, x, x_backprop. Либо «NHWC» (по умолчанию), либо «NCHW».
- is_training: логическое значение, указывающее, что операция предназначена для обучения (по умолчанию) или вывода.
Возврат:
-
Output
x_backprop: 4D- тензор для градиента по x. -
Output
Scale_backprop: 1D- тензор для градиента относительно масштаба. -
Output
offset_backprop: 1D- тензор для градиента относительно смещения. -
Output
резерв_пространство_3: неиспользуемый заполнитель, соответствующий среднему входному значению в FusedBatchNorm . -
Output
резерв_пространство_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
Общественные функции
FusedBatchНормГрад
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 )
FusedBatchНормГрад
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 )