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 | FusedBatchNormV3.Options | Attributi facoltativi per FusedBatchNormV3 |
Metodi pubblici
Uscita <U> | batchMean () Un tensore 1D per la media batch calcolata, utilizzato da TensorFlow per calcolare la media corrente. |
Uscita <U> | batchVariance () Un tensore 1D per la varianza batch calcolata, che deve essere utilizzato da TensorFlow per calcolare la varianza in esecuzione. |
static <T estende il numero, U estende il numero> FusedBatchNormV3 <T, U> | |
static FusedBatchNormV3.Options | dataFormat (String dataFormat) |
static FusedBatchNormV3.Options | epsilon (Float epsilon) |
static FusedBatchNormV3.Options | exponentialAvgFactor (Float exponentialAvgFactor) |
static FusedBatchNormV3.Options | isTraining (Boolean isTraining) |
Uscita <U> | reserveSpace1 () Un tensore 1D per la media batch calcolata, da riutilizzare nel calcolo del gradiente. |
Uscita <U> | reserveSpace2 () Un tensore 1D per la varianza batch calcolata (varianza invertita nel caso cuDNN), da riutilizzare nel calcolo del gradiente. |
Uscita <U> | reserveSpace3 () Un tensore 1D per alcuni risultati intermedi, da riutilizzare nel calcolo del gradiente per una migliore efficienza. |
Uscita <T> | y () Un tensore 4D per i dati di output. |
Metodi ereditati
Metodi pubblici
output pubblico <U> batchMean ()
Un tensore 1D per la media batch calcolata, utilizzato da TensorFlow per calcolare la media corrente.
output pubblico <U> batchVariance ()
Un tensore 1D per la varianza batch calcolata, che deve essere utilizzato da TensorFlow per calcolare la varianza in esecuzione.
public static FusedBatchNormV3 <T, U> create ( ambito ambito, operando <T> x, operando <U> scala, operando <U> offset, operando <U> media, operando <U> varianza, opzioni ... )
Metodo Factory per creare una classe che racchiude una nuova operazione FusedBatchNormV3.
Parametri
scopo | ambito attuale |
---|---|
X | Un tensore 4D per i dati di input. |
scala | Un tensore 1D per il fattore di scala, per scalare la x normalizzata. |
compensare | Un tensore 1D per offset, per passare alla x normalizzata. |
significare | Un tensore 1D per la media della popolazione. Usato solo per inferenza; deve essere vuoto per la formazione. |
varianza | Un tensore 1D per la varianza della popolazione. Usato solo per inferenza; deve essere vuoto per la formazione. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di FusedBatchNormV3
public static FusedBatchNormV3.Options dataFormat (String dataFormat)
Parametri
formato dei dati | Il formato dei dati per x e y. O "NHWC" (predefinito) o "NCHW". |
---|
public static FusedBatchNormV3.Options epsilon (Float epsilon)
Parametri
epsilon | Un piccolo numero in virgola mobile aggiunto alla varianza di x. |
---|
public static FusedBatchNormV3.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> reserveSpace1 ()
Un tensore 1D per la media batch calcolata, da riutilizzare nel calcolo del gradiente.
output pubblico <U> reserveSpace2 ()
Un tensore 1D per la varianza batch calcolata (varianza invertita nel caso cuDNN), da riutilizzare nel calcolo del gradiente.
output pubblico <U> reserveSpace3 ()
Un tensore 1D per alcuni risultati intermedi, da riutilizzare nel calcolo del gradiente per una migliore efficienza.