tensorflow::ops::FusedBatchNorm

#include <nn_ops.h>

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

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

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

Аргументы:

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

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

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

Возвращает:

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

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

::tensorflow::Output batch_mean

::tensorflow::Output batch_variance

Operation operation

::tensorflow::Output reserve_space_1

::tensorflow::Output reserve_space_2

::tensorflow::Output y

Общедоступные функции

 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
)

Общедоступные статические функции

Attrs DataFormat(
  StringPiece x
)

Attrs Epsilon(
  float x
)

Attrs ExponentialAvgFactor(
  float x
)

Attrs IsTraining(
  bool x
)