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 | FusedBatchNormGrad.Options | Attributs facultatifs pour FusedBatchNormGrad |
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
static <T étend TNumber , U étend TNumber > FusedBatchNormGrad <T, U> | |
statique FusedBatchNormGrad.Options | dataFormat (Chaîne dataFormat) |
statique FusedBatchNormGrad.Options | epsilon (epsilon flottant) |
statique FusedBatchNormGrad.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
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
public static FusedBatchNormGrad <T, U> créer ( Portée de portée , Opérande <T> yBackprop, Opérande <T> x, Opérande < TFloat32 > échelle, 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 FusedBatchNormGrad.
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 FusedBatchNormGrad
public statique FusedBatchNormGrad.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 FusedBatchNormGrad.Options epsilon (Float epsilon)
Paramètres
épsilon | Un petit nombre flottant ajouté à la variance de x. |
---|
public statique FusedBatchNormGrad.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.