тензорный поток:: опс:: FusedBatchНормГрадV3

#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-го порядка.
  • резерв_пространство_3: когда is_training имеет значение True, 1D- тензор для некоторых промежуточных результатов, который будет повторно использоваться при вычислении градиента. Когда is_training имеет значение False, будет создан фиктивный пустой Тензор .

Необязательные атрибуты (см. 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 резерв_пространство_4: неиспользуемый заполнитель, соответствующий среднему входному значению в FusedBatchNorm .
  • Output резерв_пространство_5: неиспользуемый заполнитель, соответствующий вводу отклонения в FusedBatchNorm .

Конструкторы и деструкторы

FusedBatchNormGradV3 (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2, :: tensorflow::Input reserve_space_3)
FusedBatchNormGradV3 (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2, :: tensorflow::Input reserve_space_3, const FusedBatchNormGradV3::Attrs & attrs)

Публичные атрибуты

offset_backprop
operation
reserve_space_4
reserve_space_5
scale_backprop
x_backprop

Публичные статические функции

DataFormat (StringPiece x)
Epsilon (float x)
IsTraining (bool x)

Структуры

tensorflow:: ops:: FusedBatchNormGradV3:: Attrs

Дополнительные установщики атрибутов для FusedBatchNormGradV3 .

Публичные атрибуты

offset_backprop

::tensorflow::Output offset_backprop

операция

Operation operation

резерв_пространство_4

::tensorflow::Output reserve_space_4

резерв_пространство_5

::tensorflow::Output reserve_space_5

Scale_backprop

::tensorflow::Output scale_backprop

x_backprop

::tensorflow::Output x_backprop

Общественные функции

FusedBatchНормГрадV3

 FusedBatchNormGradV3(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input y_backprop,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input reserve_space_1,
  ::tensorflow::Input reserve_space_2,
  ::tensorflow::Input reserve_space_3
)

FusedBatchНормГрадV3

 FusedBatchNormGradV3(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input y_backprop,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input reserve_space_1,
  ::tensorflow::Input reserve_space_2,
  ::tensorflow::Input reserve_space_3,
  const FusedBatchNormGradV3::Attrs & attrs
)

Публичные статические функции

Формат данных

Attrs DataFormat(
  StringPiece x
)

Эпсилон

Attrs Epsilon(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)