tensorflow :: ops :: FusedBatchNorm

#include <nn_ops.h>

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 variância da população. 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.

Construtores e Destruidores

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)

Atributos públicos

batch_mean
batch_variance
operation
reserve_space_1
reserve_space_2
y

Funções estáticas públicas

DataFormat (StringPiece x)
Epsilon (float x)
IsTraining (bool x)

Structs

tensorflow :: ops :: FusedBatchNorm :: Attrs

Configuradores de atributos opcionais para FusedBatchNorm .

Atributos públicos

batch_mean

::tensorflow::Output batch_mean

batch_variance

::tensorflow::Output batch_variance

Operação

Operation operation

reserve_space_1

::tensorflow::Output reserve_space_1

reserve_space_2

::tensorflow::Output reserve_space_2

y

::tensorflow::Output y

Funções públicas

FusedBatchNorm

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

FusedBatchNorm

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

Funções estáticas públicas

Formato de dados

Attrs DataFormat(
  StringPiece x
)

Epsilon

Attrs Epsilon(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)