тензорный поток :: ops :: FusedBatchNorm

#include <nn_ops.h>

Пакетная нормализация.

Резюме

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

Аргументы:

  • scope: объект Scope
  • x: 4D тензор для входных данных.
  • scale: 1D тензор для коэффициента масштабирования, чтобы масштабировать нормализованный x.
  • offset: 1D тензор для смещения, чтобы перейти к нормализованному x.
  • среднее: 1D тензор для среднего значения популяции. Используется только для вывода; должен быть пустым для тренировки.
  • дисперсия: одномерный тензор дисперсии совокупности. Используется только для вывода; должен быть пустым для тренировки.

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

  • эпсилон: небольшое число с плавающей запятой, добавленное к дисперсии x.
  • data_format: формат данных для x и y. Либо «NHWC» (по умолчанию), либо «NCHW».
  • is_training: логическое значение, указывающее, что операция предназначена для обучения (по умолчанию) или вывода.

Возврат:

  • Output y: 4D тензор для выходных данных.
  • Output batch_mean: одномерный тензор для вычисленного среднего значения пакета, который будет использоваться TensorFlow для вычисления текущего среднего.
  • Output batch_variance: одномерный тензор для вычисленной пакетной дисперсии, который будет использоваться TensorFlow для вычисления текущей дисперсии.
  • Output файл reserve_space_1: одномерный тензор для вычисленного среднего значения пакета, который будет повторно использован в вычислении градиента.
  • Output Reserve_space_2: 1D тензор для вычисленной дисперсии пакета (инвертированная дисперсия в случае cuDNN), который будет повторно использоваться в вычислении градиента.

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

FusedBatchNorm (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance)
FusedBatchNorm (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance, const FusedBatchNorm::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)

Структуры

tenorflow :: ops :: FusedBatchNorm :: Attrs

Необязательные установщики атрибутов для FusedBatchNorm .

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

batch_mean

::tensorflow::Output batch_mean

batch_variance

::tensorflow::Output batch_variance

операция

Operation operation

Reserve_space_1

::tensorflow::Output reserve_space_1

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

::tensorflow::Output reserve_space_2

y

::tensorflow::Output y

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

FusedBatchNorm

 FusedBatchNorm(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance
)

FusedBatchNorm

 FusedBatchNorm(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance,
  const FusedBatchNorm::Attrs & attrs
)

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

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

Attrs DataFormat(
  StringPiece x
)

Эпсилон

Attrs Epsilon(
  float x
)

ExponentialAvgFactor

Attrs ExponentialAvgFactor(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)