tensorflow::ops::FusedBatchNormV2

#include <nn_ops.h>

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

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

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

Аргументы:

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

Дополнительные атрибуты (см. Attrs ):

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

Возвращает:

  • Output y: четырехмерный тензор для выходных данных.
  • Output batch_mean: Одномерный тензор для вычисленного среднего значения пакета, который будет использоваться TensorFlow для вычисления скользящего среднего.
  • Output batch_variance: Одномерный тензор для вычисленной дисперсии пакета, который будет использоваться TensorFlow для вычисления скользящей дисперсии.
  • Output параметр reserve_space_1: одномерный тензор для вычисленного среднего значения пакета, который будет повторно использован при вычислении градиента.
  • Output параметр reserve_space_2: одномерный тензор для вычисленной дисперсии пакета (обратная дисперсия в случае 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)
IsTraining (bool x)

Структуры

tensorflow::ops::FusedBatchNormV2::Attrs

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

Общедоступные атрибуты

batch_mean 

::tensorflow::Output batch_mean

batch_variance 

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

IsTraining 

Attrs IsTraining(
  bool x
)