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

#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 Reserve_space_1: 1D- тензор для вычисленного среднего значения партии, который будет повторно использоваться при вычислении градиента.
  • Output пространство Reserve_space_2: 1D- тензор для вычисленной пакетной дисперсии (инвертированной дисперсии в случае cuDNN), который будет повторно использоваться при вычислении градиента.

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

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

Структуры

tensorflow:: ops:: FusedBatchNormV2:: Attrs

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

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

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

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

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

FusedBatchNormV2

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

FusedBatchNormV2

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

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

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

Attrs DataFormat(
  StringPiece x
)

Эпсилон

Attrs Epsilon(
  float x
)

ЭкспоненциальныйAvgFactor

Attrs ExponentialAvgFactor(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)