tensorflow :: ops :: FusedBatchNormV3
#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 de 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. -
Output
reserve_space_3: Um tensor 1D para alguns resultados intermediários, para ser reutilizado no cálculo do gradiente para melhor eficiência.
Construtores e Destruidores | |
---|---|
FusedBatchNormV3 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance) | |
FusedBatchNormV3 (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance, const FusedBatchNormV3::Attrs & attrs) |
Atributos públicos | |
---|---|
batch_mean | |
batch_variance | |
operation | |
reserve_space_1 | |
reserve_space_2 | |
reserve_space_3 | |
y |
Funções estáticas públicas | |
---|---|
DataFormat (StringPiece x) | |
Epsilon (float x) | |
IsTraining (bool x) |
Structs | |
---|---|
tensorflow :: ops :: FusedBatchNormV3 :: Attrs | Configuradores de atributos opcionais para FusedBatchNormV3 . |
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
reserve_space_3
::tensorflow::Output reserve_space_3
y
::tensorflow::Output y
Funções públicas
FusedBatchNormV3
FusedBatchNormV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance )
FusedBatchNormV3
FusedBatchNormV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV3::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 )