classe finale publique FusedBatchNormGradV3
Dégradé pour la normalisation des lots.
Notez que la taille des Tensors 4D est définie par « NHWC » ou « NCHW ». La taille des Tensors 1D correspond à la dimension C des Tensors 4D.
Classes imbriquées
| classe | FusedBatchNormGradV3.Options | Attributs facultatifs pour FusedBatchNormGradV3 | |
Méthodes publiques
| static <T étend le nombre, U étend le nombre> FusedBatchNormGradV3 <T, U> | |
| statique FusedBatchNormGradV3.Options | dataFormat (Chaîne dataFormat) |
| statique FusedBatchNormGradV3.Options | epsilon (epsilon flottant) |
| statique FusedBatchNormGradV3.Options | isTraining (isTraining booléen) |
| Sortie <U> | offsetBackprop () Un tenseur 1D pour le dégradé par rapport au décalage. |
| Sortie <U> | réserveEspace4 () Espace réservé inutilisé pour correspondre à l’entrée moyenne dans FusedBatchNorm. |
| Sortie <U> | réserveEspace5 () Espace réservé inutilisé pour correspondre à l’entrée de variance dans FusedBatchNorm. |
| Sortie <U> | échelleBackprop () Un tenseur 1D pour le dégradé par rapport à l'échelle. |
| Sortie <T> | xBackprop () Un tenseur 4D pour le dégradé par rapport à x. |
Méthodes héritées
Méthodes publiques
public static FusedBatchNormGradV3 <T, U> créer ( Portée de la portée , Opérande <T> yBackprop, Opérande <T> x, Échelle de l' Opérande <Float>, Opérande <U> ReserveSpace1, Opérande <U> ReserveSpace2, Opérande <U> ReserveSpace3, Options ... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération FusedBatchNormGradV3.
Paramètres
| portée | portée actuelle |
|---|---|
| yBackprop | Un tenseur 4D pour le gradient par rapport à y. |
| X | Un Tensor 4D pour les données d'entrée. |
| échelle | Un tenseur 1D pour le facteur de mise à l'échelle, pour mettre à l'échelle le x normalisé. |
| réserveEspace1 | Lorsque is_training est True, un tenseur 1D pour le lot calculé doit être réutilisé dans le calcul du gradient. Lorsque is_training est False, un tenseur 1D pour la population doit être réutilisé dans le calcul du gradient du 1er et du 2ème ordre. |
| réserveEspace2 | Lorsque is_training est True, un tenseur 1D pour la variance par lot calculée (variance inversée dans le cas cuDNN) à réutiliser dans le calcul du gradient. Lorsque is_training est False, un tenseur 1D pour la variance de la population doit être réutilisé dans le calcul du gradient du 1er et du 2e ordre. |
| réserveEspace3 | Lorsque is_training est True, un tenseur 1D pour certains résultats intermédiaires à réutiliser dans le calcul du gradient. Lorsque is_training est False, un Tensor vide factice sera créé. |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de FusedBatchNormGradV3
public statique FusedBatchNormGradV3.Options dataFormat (String dataFormat)
Paramètres
| format de données | Le format de données pour y_backprop, x, x_backprop. Soit « NHWC » (par défaut) soit « NCHW ». |
|---|
public statique FusedBatchNormGradV3.Options epsilon (Float epsilon)
Paramètres
| épsilon | Un petit nombre flottant ajouté à la variance de x. |
|---|
public statique FusedBatchNormGradV3.Options isTraining (booléen isTraining)
Paramètres
| estFormation | Une valeur booléenne pour indiquer que l'opération est destinée à la formation (par défaut) ou à l'inférence. |
|---|
sortie publique <U> reserveSpace4 ()
Espace réservé inutilisé pour correspondre à l’entrée moyenne dans FusedBatchNorm.
sortie publique <U> reserveSpace5 ()
Espace réservé inutilisé pour correspondre à l’entrée de variance dans FusedBatchNorm.