tensorflow::ops::FusedBatchNormGradV2

#include <nn_ops.h>

Градиент для пакетной нормализации.

Краткое описание

Обратите внимание, что размер 4D тензоров определяется либо «NHWC», либо «NCHW». Размер 1D тензоров соответствует размеру C 4D тензоров.

Аргументы:

• scope: Объект Scope
• y_backprop: 4D Tensor для градиента относительно y.
• x: 4D Tensor для входных данных.
• scale: 1D Tensor для коэффициента масштабирования, для масштабирования нормализованного x.
• reserve_space_1: При is_training = True, 1D Tensor для вычисленного среднего значения пакета, которое будет повторно использовано в вычислении градиента. При is_training = False, 1D Tensor для популяционного среднего значения, которое будет повторно использовано как в вычислении градиента первого порядка, так и второго порядка.
• reserve_space_2: При is_training = True, 1D Tensor для вычисленной дисперсии пакета (обратной дисперсии в случае cuDNN), которая будет повторно использована в вычислении градиента. При is_training = False, 1D Tensor для популяционной дисперсии, которая будет повторно использована как в вычислении градиента первого порядка, так и второго порядка.

Необязательные атрибуты (см. Attrs):

• epsilon: Малое число с плавающей точкой, добавляемое к дисперсии x.
• data_format: Формат данных для y_backprop, x, x_backprop. Либо "NHWC" (по умолчанию), либо "NCHW".
• is_training: Значение bool, указывающее, предназначена ли операция для обучения (по умолчанию) или для инференции.

Возвращает:

• Output x_backprop: 4D Tensor для градиента относительно x.
• Output scale_backprop: 1D Tensor для градиента относительно scale.
• Output offset_backprop: 1D Tensor для градиента относительно offset.
• Output reserve_space_3: Неиспользуемый плейсхолдер для соответствия входному среднему в FusedBatchNorm.
• Output reserve_space_4: Неиспользуемый плейсхолдер для соответствия входной дисперсии в FusedBatchNorm.

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

::tensorflow::Output offset_backprop

Operation operation

::tensorflow::Output reserve_space_3

::tensorflow::Output reserve_space_4

::tensorflow::Output scale_backprop

::tensorflow::Output x_backprop

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

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

 FusedBatchNormGradV2(
  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 FusedBatchNormGradV2::Attrs & attrs
)

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

Attrs DataFormat(
  StringPiece x
)

Attrs Epsilon(
  float x
)

Attrs IsTraining(
  bool x
)