tensorflow::ops::FusedBatchNormV2

#include <nn_ops.h>

Нормализация по батчу.

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

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

Аргументы:

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

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

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

Возвращаемое значение:

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

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

::tensorflow::Output batch_mean

::tensorflow::Output batch_variance

Operation operation

::tensorflow::Output reserve_space_1

::tensorflow::Output reserve_space_2

::tensorflow::Output y

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

 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
)

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

Attrs DataFormat(
  StringPiece x
)

Attrs Epsilon(
  float x
)

Attrs ExponentialAvgFactor(
  float x
)

Attrs IsTraining(
  bool x
)