FusedBatchNormGradV3

public final class FusedBatchNormGradV3

Gradiente per la normalizzazione batch.

Notare che le dimensioni dei tensori 4D sono definite da "NHWC" o "NCHW". La dimensione dei tensori 1D corrisponde alla dimensione C dei tensori 4D.

Classi annidate

classe FusedBatchNormGradV3.Options Attributi facoltativi per FusedBatchNormGradV3

Metodi pubblici

static <T estende il numero, U estende il numero> FusedBatchNormGradV3 <T, U>
create ( Scope scope, Operand <T> yBackprop, Operand <T> x, Operand <Float> scale, Operand <U> reserveSpace1, Operand <U> reserveSpace2, Operand <U> reserveSpace3, Options ... options)
Metodo Factory per creare una classe che racchiude una nuova operazione FusedBatchNormGradV3.
static FusedBatchNormGradV3.Options
dataFormat (String dataFormat)
static FusedBatchNormGradV3.Options
epsilon (Float epsilon)
static FusedBatchNormGradV3.Options
isTraining (Boolean isTraining)
Uscita <U>
offsetBackprop ()
Un tensore 1D per il gradiente rispetto all'offset.
Uscita <U>
reserveSpace4 ()
Segnaposto inutilizzato per abbinare l'input medio in FusedBatchNorm.
Uscita <U>
reserveSpace5 ()
Segnaposto inutilizzato per abbinare l'input della varianza in FusedBatchNorm.
Uscita <U>
scaleBackprop ()
Un tensore 1D per il gradiente rispetto alla scala.
Uscita <T>
xBackprop ()
Un tensore 4D per il gradiente rispetto a x.

Metodi ereditati

Metodi pubblici

public static FusedBatchNormGradV3 <T, U> create ( Scope scope, Operand <T> yBackprop, Operand <T> x, Operand <Float> scale, Operand <U> reserveSpace1, Operand <U> reserveSpace2, Operand <U> reserveSpace3, Opzioni ... opzioni)

Metodo Factory per creare una classe che racchiude una nuova operazione FusedBatchNormGradV3.

Parametri
scopo ambito attuale
yBackprop Un tensore 4D per il gradiente rispetto a y.
X Un tensore 4D per i dati di input.
scala Un tensore 1D per fattore di scala, per scalare la x normalizzata.
reserveSpace1 Quando is_training è True, un tensore 1D per il batch calcolato significa che deve essere riutilizzato nel calcolo del gradiente. Quando is_training è False, un tensore 1D per la popolazione significa che deve essere riutilizzato sia nel calcolo del gradiente di 1 ° che di 2 ° ordine.
reserveSpace2 Quando is_training è True, un tensore 1D per la varianza batch calcolata (varianza invertita nel caso cuDNN) da riutilizzare nel calcolo del gradiente. Quando is_training è False, un tensore 1D per la varianza della popolazione da riutilizzare sia nel calcolo del gradiente di 1 ° che di 2 ° ordine.
reserveSpace3 Quando is_training è True, un tensore 1D per alcuni risultati intermedi da riutilizzare nel calcolo del gradiente. Quando is_training è False, verrà creato un tensore vuoto fittizio.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di FusedBatchNormGradV3

public static FusedBatchNormGradV3.Options dataFormat (String dataFormat)

Parametri
formato dei dati Il formato dei dati per y_backprop, x, x_backprop. O "NHWC" (predefinito) o "NCHW".

public static FusedBatchNormGradV3.Options epsilon (Float epsilon)

Parametri
epsilon Un piccolo numero in virgola mobile aggiunto alla varianza di x.

public static FusedBatchNormGradV3.Options isTraining (Boolean isTraining)

Parametri
isTraining Un valore bool per indicare che l'operazione è per l'addestramento (impostazione predefinita) o l'inferenza.

output pubblico <U> offsetBackprop ()

Un tensore 1D per il gradiente rispetto all'offset.

output pubblico <U> reserveSpace4 ()

Segnaposto inutilizzato per abbinare l'input medio in FusedBatchNorm.

output pubblico <U> reserveSpace5 ()

Segnaposto inutilizzato per abbinare l'input della varianza in FusedBatchNorm.

output pubblico <U> scaleBackprop ()

Un tensore 1D per il gradiente rispetto alla scala.

output pubblico <T> xBackprop ()

Un tensore 4D per il gradiente rispetto a x.