Normalização em lote.
Observe que o tamanho dos tensores 4D é definido por "NHWC" ou "NCHW". O tamanho dos tensores 1D corresponde à dimensão C dos tensores 4D.
Classes aninhadas
aula | FusedBatchNormV3.Options | Atributos opcionais para FusedBatchNormV3 |
Métodos públicos
Saída <U> | batchMean () Um tensor 1D para a média de lote computada, a ser usado pelo TensorFlow para calcular a média de execução. |
Saída <U> | batchVariance () Um tensor 1D para a variação de lote computada, a ser usado pelo TensorFlow para calcular a variação em execução. |
estático <T estende o número, U estende o número> FusedBatchNormV3 <T, U> | |
estático FusedBatchNormV3.Options | dataFormat (String dataFormat) |
estático FusedBatchNormV3.Options | epsilon (Float epsilon) |
estático FusedBatchNormV3.Options | exponencialAvgFactor (Float exponencialAvgFactor) |
estático FusedBatchNormV3.Options | isTraining (booleano isTraining) |
Saída <U> | reserveSpace1 () Um tensor 1D para a média de lote computada, para ser reutilizado no cálculo do gradiente. |
Saída <U> | reserveSpace2 () Um Tensor 1D para a variância de lote computada (variância invertida no caso cuDNN), a ser reutilizada no cálculo do gradiente. |
Saída <U> | reserveSpace3 () Um Tensor 1D para alguns resultados intermediários, para ser reutilizado no cálculo do gradiente para melhor eficiência. |
Saída <T> | e () Um tensor 4D para dados de saída. |
Métodos Herdados
Métodos públicos
Public Output <U> batchMean ()
Um tensor 1D para a média de lote computada, a ser usado pelo TensorFlow para calcular a média de execução.
Public Output <U> batchVariance ()
Um tensor 1D para a variação de lote computada, a ser usado pelo TensorFlow para calcular a variação em execução.
public static FusedBatchNormV3 <T, U> create ( Escopo do escopo, Operando <T> x, Escala do Operando <U>, Deslocamento do Operando <U>, Média do Operando <U>, Variação do Operando <U>, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação FusedBatchNormV3.
Parâmetros
alcance | escopo atual |
---|---|
x | Um tensor 4D para dados de entrada. |
escala | Um tensor 1D para fator de escala, para escalar o x normalizado. |
Deslocamento | Um tensor 1D para deslocamento, para mudar para o x normalizado. |
quer dizer | Um tensor 1D para média populacional. Usado apenas para inferência; deve estar vazio para treinamento. |
variação | Um tensor 1D para variação populacional. Usado apenas para inferência; deve estar vazio para treinamento. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância do FusedBatchNormV3
public static FusedBatchNormV3.Options dataFormat (String dataFormat)
Parâmetros
formato de dados | O formato de dados para x e y. Ou "NHWC" (padrão) ou "NCHW". |
---|
público estático FusedBatchNormV3.Options epsilon (Float epsilon)
Parâmetros
épsilon | Um pequeno número float adicionado à variância de x. |
---|
public static FusedBatchNormV3.Options isTraining (Boolean isTraining)
Parâmetros
isTraining | Um valor bool para indicar que a operação é para treinamento (padrão) ou inferência. |
---|
Public Output <U> reserveSpace1 ()
Um tensor 1D para a média de lote computada, para ser reutilizado no cálculo do gradiente.
Public Output <U> reserveSpace2 ()
Um Tensor 1D para a variância de lote computada (variância invertida no caso cuDNN), a ser reutilizada no cálculo do gradiente.
Public Output <U> reserveSpace3 ()
Um Tensor 1D para alguns resultados intermediários, para ser reutilizado no cálculo do gradiente para melhor eficiência.