Gradiente para normalização de 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 | FusedBatchNormGradV3.Options | Atributos opcionais para FusedBatchNormGradV3 |
Métodos Públicos
static <T estende o número, U estende o número> FusedBatchNormGradV3 <T, U> | |
estático FusedBatchNormGradV3.Options | dataFormat (String dataFormat) |
estático FusedBatchNormGradV3.Options | épsilon (épsilon flutuante) |
estático FusedBatchNormGradV3.Options | isTraining (booleano isTraining) |
Saída <U> | deslocamentoBackprop () Um tensor 1D para o gradiente em relação ao deslocamento. |
Saída <U> | reservaEspaço4 () Espaço reservado não utilizado para corresponder à entrada média em FusedBatchNorm. |
Saída <U> | reservarEspaço5 () Espaço reservado não utilizado para corresponder à entrada de variação no FusedBatchNorm. |
Saída <U> | escalaBackprop () Um Tensor 1D para o gradiente em relação à escala. |
Saída <T> | xBackprop () Um tensor 4D para o gradiente em relação a x. |
Métodos herdados
Métodos Públicos
public static FusedBatchNormGradV3 <T, U> create ( Escopo escopo , Operando <T> yBackprop, Operando <T> x, Operando <Float> escala, Operando <U> reserveSpace1, Operando <U> reserveSpace2, Operando <U> reserveSpace3, Opções ... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação FusedBatchNormGradV3.
Parâmetros
escopo | escopo atual |
---|---|
yBackprop | Um Tensor 4D para o gradiente em relação a y. |
x | Um Tensor 4D para dados de entrada. |
escala | Um Tensor 1D para fator de escala, para escalar o x normalizado. |
reservarEspaço1 | Quando is_training é True, um Tensor 1D para o lote computado significa ser reutilizado no cálculo do gradiente. Quando is_training é False, um Tensor 1D para a população deve ser reutilizado no cálculo do gradiente de 1ª e 2ª ordem. |
reservaSpace2 | Quando is_training é True, um Tensor 1D para a variação do lote computada (variância invertida no caso cuDNN) a ser reutilizado no cálculo do gradiente. Quando is_training é False, um Tensor 1D para a variância da população a ser reutilizado no cálculo do gradiente de 1ª e 2ª ordem. |
reservaSpace3 | Quando is_training é True, um Tensor 1D para alguns resultados intermediários a serem reutilizados no cálculo do gradiente. Quando is_training for False, um Tensor vazio fictício será criado. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância do FusedBatchNormGradV3
público estático FusedBatchNormGradV3.Options dataFormat (String dataFormat)
Parâmetros
formato de dados | O formato de dados para y_backprop, x, x_backprop. Ou "NHWC" (padrão) ou "NCHW". |
---|
público estático FusedBatchNormGradV3.Options epsilon (Float épsilon)
Parâmetros
épsilon | Um pequeno número flutuante adicionado à variância de x. |
---|
public static FusedBatchNormGradV3.Options isTraining (Boolean isTraining)
Parâmetros
isTraining | Um valor bool para indicar que a operação é para treinamento (padrão) ou inferência. |
---|
Saída pública <U> reserveSpace4 ()
Espaço reservado não utilizado para corresponder à entrada média em FusedBatchNorm.
Saída pública <U> reserveSpace5 ()
Espaço reservado não utilizado para corresponder à entrada de variação no FusedBatchNorm.