FusedBatchNormGradV3

Clase public final FusedBatchNormGradV3

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 Los atributos opcionales para FusedBatchNormGradV3

Métodos públicos

estática <T se extiende Número, U se extiende Número> 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 envuelva una nueva operación FusedBatchNormGradV3.
estáticas FusedBatchNormGradV3.Options
dataFormat (String dataFormat)
estáticas FusedBatchNormGradV3.Options
epsilon (epsilon Float)
estáticas FusedBatchNormGradV3.Options
isTraining (Boolean 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> 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 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 calculada del lote 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á 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".

estáticas pública FusedBatchNormGradV3.Options épsilon (épsilon Float)

Parámetros
épsilon Un pequeño número flotante agregado a la varianza de x.

public static FusedBatchNormGradV3.Options isTraining (booleano isTraining)

Parámetros
está entrenando Un valor bool para indicar que la operación es para entrenamiento (predeterminado) o inferencia.

pública de salida <U> offsetBackprop ()

Un tensor 1D para el gradiente con respecto al desplazamiento.

pública de salida <U> reserveSpace4 ()

Marcador de posición no utilizado para que coincida con la entrada media en FusedBatchNorm.

pública de salida <U> reserveSpace5 ()

Marcador de posición no utilizado para que coincida con la entrada de variación en FusedBatchNorm.

pública de salida <U> scaleBackprop ()

Un tensor 1D para el gradiente con respecto a la escala.

pública de salida <T> xBackprop ()

Un tensor 4D para el gradiente con respecto ax.