Gradiente para normalización por 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 | FusionadoBatchNormGradV3.Options | Atributos opcionales para FusedBatchNormGradV3 |
Métodos públicos
estático <T extiende el número, U extiende el número> FusedBatchNormGradV3 <T, U> | |
Estático FusedBatchNormGradV3.Options | formato de datos (formato de datos de cadena) |
Estático FusedBatchNormGradV3.Options | épsilon (épsilon flotante) |
Estático FusedBatchNormGradV3.Options | isTraining (booleano isTraining) |
Salida <U> | offsetBackprop () Un tensor 1D para el gradiente con respecto al desplazamiento. |
Salida <U> | reservarEspacio4 () Marcador de posición no utilizado para coincidir con la entrada media en FusedBatchNorm. |
Salida <U> | reservarEspacio5 () Marcador de posición no utilizado para coincidir con 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
público estático FusedBatchNormGradV3 <T, U> crear ( Alcance alcance, Operando <T> yBackprop, Operando <T> x, Operando <Float> escala, Operando <U> reserveSpace1, Operando <U> reserveSpace2, Operando <U> reserveSpace3, 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. |
reservarEspacio1 | Cuando is_training es True, un tensor 1D para el lote calculado significa que se reutilizará en el cálculo del gradiente. Cuando is_training es False, un tensor 1D para la población significa que se reutilizará en el cálculo de gradiente de primer y segundo orden. |
reservarEspacio2 | Cuando is_training es True, un tensor 1D para la varianza del lote calculada (varianza invertida en el caso de cuDNN) se reutilizará en el cálculo del gradiente. Cuando is_training es False, se reutilizará un tensor 1D para la varianza de la población en el cálculo del gradiente de primer y segundo orden. |
reservarEspacio3 | Cuando is_training es True, se reutiliza un tensor 1D para algunos resultados intermedios en el cálculo del gradiente. Cuando is_training es False, se creará un tensor vacío ficticio. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de FusedBatchNormGradV3
público estático FusedBatchNormGradV3.Options dataFormat (String dataFormat)
Parámetros
formato de datos | El formato de datos para y_backprop, x, x_backprop. Ya sea "NHWC" (predeterminado) o "NCHW". |
---|
público estático FusedBatchNormGradV3.Options épsilon (Float épsilon)
Parámetros
épsilon | Un pequeño número flotante añadido a la varianza de x. |
---|
público estático FusedBatchNormGradV3.Options isTraining (booleano isTraining)
Parámetros
está entrenando | Un valor bool para indicar que la operación es para entrenamiento (predeterminado) o inferencia. |
---|
Salida pública <U> reserveSpace4 ()
Marcador de posición no utilizado para coincidir con la entrada media en FusedBatchNorm.
Salida pública <U> reserveSpace5 ()
Marcador de posición no utilizado para coincidir con la entrada de varianza en FusedBatchNorm.