tensorflow :: ops :: FusedBatchNorm

Normalização em lote.

Resumo

Observe que o tamanho dos tensores 4D é definido por "NHWC" ou "NCHW". O tamanho dos tensores 1D corresponde à dimensão C dos tensores 4D.

Argumentos:

• escopo: um objeto Scope
• x: um tensor 4D para dados de entrada.
• escala: Um tensor 1D para fator de escala, para dimensionar o x normalizado.
• deslocamento: Um tensor 1D para deslocamento, para deslocar para o x normalizado.
• média: Um tensor 1D para a média da população. Usado apenas para inferência; deve estar vazio para treinamento.
• variância: Um tensor 1D para variação populacional. Usado apenas para inferência; deve estar vazio para treinamento.

Atributos opcionais (consulte Attrs ):

• epsilon: Um pequeno número flutuante adicionado à variância de x.
• data_format: o formato de dados para x e y. "NHWC" (padrão) ou "NCHW".
• is_training: um valor booleano para indicar que a operação é para treinamento (padrão) ou inferência.

Retorna:

• Output y: um tensor 4D para dados de saída.
• Output batch_mean: um tensor 1D para a média do lote calculada, a ser usado pelo TensorFlow para calcular a média de execução
• Output batch_variance: um tensor 1D para a variação de lote calculada, a ser usado pelo TensorFlow para calcular a variação em execução.
• Output reserve_space_1: um tensor 1D para a média de lote calculada, para ser reutilizado no cálculo do gradiente.
• Output reserve_space_2: Um tensor 1D para a variância de lote calculada (variância invertida no caso cuDNN), para ser reutilizado no cálculo do gradiente.

Atributos públicos

::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 IsTraining(
  bool x
)