Gradiente para la normalización de lotes.
Tenga en cuenta que el tamaño de los tensores 4D está definido por "NHWC" o "NCHW". El tamaño de los tensores 1D coincide con la dimensión C de los tensores 4D.
Clases anidadas
clase | FusedBatchNormGradV3.Opciones | Atributos opcionales para FusedBatchNormGradV3 |
Métodos públicos
estático <T extiende Número, U extiende Número> FusedBatchNormGradV3 <T, U> | create ( Ámbito ámbito, Operando <T> yBackprop, Operando <T> x, Operando <Flotante> escala, Operando <U> reservarEspacio1, Operando <U> reservarEspacio2, Operando <U> reservarEspacio3, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación FusedBatchNormGradV3. |
FusedBatchNormGradV3.Options estático | formato de datos (cadena de formato de datos) |
FusedBatchNormGradV3.Options estático | épsilon (épsilon flotante) |
FusedBatchNormGradV3.Options estático | isTraining (booleano isTraining) |
Salida <U> | desplazamientoBackprop () Un tensor 1D para el gradiente con respecto al desplazamiento. |
Salida <U> | reservaEspacio4 () Marcador de posición no utilizado para que coincida con la entrada media en FusedBatchNorm. |
Salida <U> | reservaEspacio5 () Marcador de posición sin usar para hacer coincidir la entrada de varianza en FusedBatchNorm. |
Salida <U> | escalaBackprop () Un tensor 1D para el gradiente con respecto a la escala. |
Salida <T> | xBackprop () Un tensor 4D para el gradiente con respecto a x. |
Métodos Heredados
Métodos públicos
public static FusedBatchNormGradV3 <T, U> create ( Ámbito ámbito, Operando <T> yBackprop, Operando <T> x, Operando <Flotante> escala, Operando <U> reservarEspacio1, Operando <U> reservarEspacio2, Operando <U> reservarEspacio3, Opciones ... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación FusedBatchNormGradV3.
Parámetros
alcance | alcance actual |
---|---|
yBackprop | Un tensor 4D para el gradiente con respecto a y. |
X | Un tensor 4D para datos de entrada. |
escala | Un tensor 1D para factor de escala, para escalar la x normalizada. |
reservaEspacio1 | Cuando is_training es verdadero, un tensor 1D para el lote calculado significa que se reutilizará en el cálculo de gradiente. Cuando is_training es falso, se reutiliza un tensor 1D para la media de población en el cálculo de gradiente de primer y segundo orden. |
reservaEspacio2 | Cuando is_training es True, un tensor 1D para la varianza del lote calculada (varianza invertida en el caso de cuDNN) para ser reutilizado en el cálculo de gradiente. Cuando is_training es falso, se reutiliza un tensor 1D para la varianza de la población en el cálculo de gradientes de primer y segundo orden. |
reservaEspacio3 | Cuando is_training es True, se reutiliza un tensor 1D para algunos resultados intermedios en el cálculo de gradientes. Cuando is_training es falso, se creará un tensor vacío ficticio. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de FusedBatchNormGradV3
public static FusedBatchNormGradV3.Options dataFormat (String dataFormat)
Parámetros
formato de datos | El formato de datos para y_backprop, x, x_backprop. O "NHWC" (predeterminado) o "NCHW". |
---|
público estático FusedBatchNormGradV3.Options épsilon (épsilon flotante)
Parámetros
épsilon | Un pequeño número flotante agregado a la varianza de x. |
---|
public static FusedBatchNormGradV3.Options isTraining (Boolean isTraining)
Parámetros
está entrenando | Un valor booleano para indicar que la operación es para entrenamiento (predeterminado) o inferencia. |
---|
Salida pública <U> reserveSpace4 ()
Marcador de posición no utilizado para que coincida con la entrada media en FusedBatchNorm.
Salida pública <U> reserveSpace5 ()
Marcador de posición sin usar para hacer coincidir la entrada de varianza en FusedBatchNorm.