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

#include <nn_ops.h>

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

Краткое содержание

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

Аргументы:

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

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

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

Возврат:

  • Output y: 4D- тензор для выходных данных.
  • Output пакет_mean: 1D- тензор для вычисленного среднего значения пакета, который будет использоваться TensorFlow для вычисления текущего среднего значения.
  • Output пакетная дисперсия: 1D- тензор для вычисленной пакетной дисперсии, который будет использоваться TensorFlow для вычисления текущей дисперсии.
  • Output резерв_пространство_1: 1D- тензор для вычисленного среднего значения партии, который будет повторно использоваться при вычислении градиента.
  • 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)
IsTraining (bool x)

Структуры

tensorflow:: ops:: FusedBatchNorm:: Attrs

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

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

пакетное_среднее 

::tensorflow::Output batch_mean

пакетная_вариация 

::tensorflow::Output batch_variance

операция 

Operation operation

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

::tensorflow::Output reserve_space_1

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

::tensorflow::Output reserve_space_2

й 

::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
)

IsTraining 

Attrs IsTraining(
  bool x
)