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 extends Number, U extends Number> FusedBatchNormGradV3 <T, U> | |
static FusedBatchNormGradV3.Options | dataFormat (String dataFormat) |
static FusedBatchNormGradV3.Options | |
static FusedBatchNormGradV3.Options | isTraining (Boolean isTraining) |
Saída <U> | offsetBackprop () Um tensor 1D para o gradiente em relação ao deslocamento. |
Saída <U> | reserveSpace4 () Espaço reservado não usado para corresponder à entrada média em FusedBatchNorm. |
Saída <U> | reserveSpace5 () Espaço reservado não usado para corresponder à entrada de variação em FusedBatchNorm. |
Saída <U> | scaleBackprop () 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> criar ( Escopo de escopo, Operando <T> yBackprop, Operando <T> x, Escala de Operando <Float>, Operando <U> reserveSpace1, Operando <U> reserveSpace2, Operando <U> reserveSpace3, Opções ... opções)
Método de fábrica para criar uma classe que envolve uma nova operação FusedBatchNormGradV3.
Parâmetros
alcance | 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 dimensionar o x normalizado. |
reserveSpace1 | Quando is_training for True, um tensor 1D para o lote calculado deve ser reutilizado na computação de gradiente. Quando is_training é False, um tensor 1D para a população significa que deve ser reutilizado no cálculo do gradiente de 1ª e 2ª ordem. |
reserveSpace2 | Quando is_training for True, um tensor 1D para a variância calculada do lote (variância invertida no caso cuDNN) a ser reutilizado no cálculo do gradiente. Quando is_training for False, um tensor 1D para a variância da população a ser reutilizado no cálculo do gradiente de 1ª e 2ª ordem. |
reserveSpace3 | Quando is_training for True, um tensor 1D para alguns resultados intermediários a serem reutilizados na computação de gradiente. Quando is_training for False, um tensor vazio simulado será criado. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de FusedBatchNormGradV3
public static FusedBatchNormGradV3.Options dataFormat (String dataFormat)
Parâmetros
formato de dados | O formato de dados para y_backprop, x, x_backprop. "NHWC" (padrão) ou "NCHW". |
---|
public static FusedBatchNormGradV3.Options epsilon (Float epsilon)
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 booleano para indicar que a operação é para treinamento (padrão) ou inferência. |
---|
public Output <U> reserveSpace4 ()
Espaço reservado não utilizado para corresponder à entrada média em FusedBatchNorm.
public Output <U> reserveSpace5 ()
Espaço reservado não usado para corresponder à entrada de variação em FusedBatchNorm.