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
classe | FusedBatchNormGradV3.Options | Atributos opcionais para FusedBatchNormGradV3 |
Métodos públicos
estático <T estende o número, U estende o número> FusedBatchNormGradV3 <T, U> | |
static FusedBatchNormGradV3.Options | dataFormat (String dataFormat) |
static FusedBatchNormGradV3.Options | epsilon (Float epsilon) |
static FusedBatchNormGradV3.Options | isTraining (booleano 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 utilizado para corresponder à entrada média em FusedBatchNorm. |
Saída <U> | reserveSpace5 () Espaço reservado não utilizado para corresponder à entrada de variação em 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 do 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 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. |
régua | Um tensor 1D para fator de escala, para escalar o x normalizado. |
reserveSpace1 | Quando is_training for True, um Tensor 1D para o lote calculado significa ser reutilizado no cálculo de gradiente. Quando is_training for False, um Tensor 1D para a população significa ser reutilizado no cálculo do gradiente de 1ª e 2ª ordem. |
reserveSpace2 | Quando is_training for True, um Tensor 1D para a variação de lote computada (variação invertida no caso cuDNN) será reutilizado no cálculo de gradiente. Quando is_training for False, um Tensor 1D para a variação 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 serem reutilizados no cálculo de 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 de 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 epsilon)
Parâmetros
épsilon | Um pequeno número float 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. |
---|
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 utilizado para corresponder à entrada de variação em FusedBatchNorm.