FusedBatchNormGradV3

clase final pública 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 Atributos opcionales para FusedBatchNormGradV3

Métodos públicos

estático <T extiende Número, U 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.
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> offsetBackprop ()

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

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.

Salida pública <U> scaleBackprop ()

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

Salida pública <T> xBackprop ()

Un tensor 4D para el gradiente con respecto ax.