Gradiente para 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.Options | Atributos opcionales para FusedBatchNormGradV3 |
Métodos públicos
estático <T extiende Número, U extiende Número> FusedBatchNormGradV3 <T, U> | |
static FusedBatchNormGradV3.Options | dataFormat (String dataFormat) |
static FusedBatchNormGradV3.Options | épsilon (flotador épsilon) |
static FusedBatchNormGradV3.Options | isTraining (booleano isTraining) |
Salida <U> | offsetBackprop () Un tensor 1D para el gradiente con respecto al desplazamiento. |
Salida <U> | reserveSpace4 () Marcador de posición no utilizado para que coincida con la entrada media en FusedBatchNorm. |
Salida <U> | reserveSpace5 () Marcador de posición no utilizado para que coincida con la entrada de variación en FusedBatchNorm. |
Salida <U> | scaleBackprop () Un tensor 1D para el gradiente con respecto a la escala. |
Salida <T> | xBackprop () Un tensor 4D para el gradiente con respecto ax. |
Métodos heredados
Métodos públicos
public static FusedBatchNormGradV3 <T, U> create ( alcance de alcance, operando <T> yBackprop, operando <T> x, escala de operando <Float>, operando <U> reserveSpace1, operando <U> reserveSpace2, operando <U> reserveSpace3, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación FusedBatchNormGradV3.
Parámetros
alcance | alcance actual |
---|---|
yBackprop | Un tensor 4D para el gradiente con respecto ay. |
X | Un tensor 4D para datos de entrada. |
escala | Un tensor 1D para el factor de escala, para escalar la x normalizada. |
reserveSpace1 | Cuando is_training es True, un tensor 1D para la media del lote calculada se reutilizará en el cálculo del gradiente. Cuando is_training es False, un tensor 1D para la media de la población se reutilizará en el cálculo del gradiente de primer y segundo orden. |
reserveSpace2 | Cuando is_training es Verdadero, 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, un tensor 1D para la varianza de la población se reutilizará en el cálculo del gradiente de primer y segundo orden. |
reserveSpace3 | Cuando is_training es True, un tensor 1D para algunos resultados intermedios que se reutilizarán 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
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". |
---|
public static FusedBatchNormGradV3.Options epsilon (Float epsilon)
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 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 que coincida con la entrada media en FusedBatchNorm.
Salida pública <U> reserveSpace5 ()
Marcador de posición no utilizado para que coincida con la entrada de variación en FusedBatchNorm.