tensorflow::ops::FusedBatchNormGrad
#include <nn_ops.h>Gradient pour la normalisation par lots.
Résumé
Notez que la taille des tenseurs 4D est définie par « NHWC » ou « NCHW ». La taille des tenseurs 1D correspond à la dimension C des tenseurs 4D.
Arguments :
- portée : un objet Scope
- y_backprop : Un tenseur 4D pour le gradient par rapport à y.
- x : Un tenseur 4D pour les données d'entrée.
- scale : Un tenseur 1D pour le facteur d'échelle, afin de mettre à l'échelle le x normalisé.
- reserve_space_1 : Si is_training est vrai, un tenseur 1D contenant la moyenne du lot calculé sera réutilisé pour le calcul du gradient. Si is_training est faux, un tenseur 1D contenant la moyenne de la population sera réutilisé pour le calcul des gradients d'ordre 1 et 2.
- reserve_space_2 : Si is_training est vrai, un tenseur 1D contenant la variance du lot calculé (variance inverse dans le cas de cuDNN) sera réutilisé pour le calcul du gradient. Si is_training est faux, un tenseur 1D contenant la variance de la population sera réutilisé pour le calcul des gradients d'ordre 1 et 2.
Attributs optionnels (voir Attrs ) :
- epsilon : Un petit nombre flottant ajouté à la variance de x.
- data_format : Format des données pour y_backprop, x et x_backprop. Soit « NHWC » (par défaut), soit « NCHW ».
- is_training : une valeur booléenne indiquant que l’opération est destinée à l’entraînement (par défaut) ou à l’inférence.
Retours :
-
Outputx_backprop : Un tenseur 4D pour le gradient par rapport à x. - scale_backprop
Output: un tenseur 1D pour le gradient par rapport à l’échelle. -
Outputoffset_backprop : Un tenseur 1D pour le gradient par rapport au décalage. -
Outputreserve_space_3 : Espace réservé inutilisé pour correspondre à l’entrée moyenne dans FusedBatchNorm . -
Outputreserve_space_4 : Espace réservé inutilisé pour correspondre à l’entrée de variance dans FusedBatchNorm .
Constructeurs et destructeurs | |
|---|---|
FusedBatchNormGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2) | |
FusedBatchNormGrad (const :: tensorflow::Scope & scope, :: tensorflow::Input y_backprop, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input reserve_space_1, :: tensorflow::Input reserve_space_2, const FusedBatchNormGrad::Attrs & attrs) |
attributs publics | |
|---|---|
offset_backprop | |
operation | |
reserve_space_3 | |
reserve_space_4 | |
scale_backprop | |
x_backprop | |
fonctions statiques publiques | |
|---|---|
DataFormat (StringPiece x) | |
Epsilon (float x) | |
IsTraining (bool x) | |
Structures | |
|---|---|
| tensorflow::ops::FusedBatchNormGrad::Attrs | Définiteurs d'attributs optionnels pour FusedBatchNormGrad . |
attributs publics
offset_backprop
::tensorflow::Output offset_backprop
opération
Operation operation
réserve_espace_3
::tensorflow::Output reserve_space_3
réserve_espace_4
::tensorflow::Output reserve_space_4
scale_backprop
::tensorflow::Output scale_backprop
x_backprop
::tensorflow::Output x_backprop
fonctions publiques
FusedBatchNormGrad
FusedBatchNormGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input y_backprop, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input reserve_space_1, ::tensorflow::Input reserve_space_2 )
FusedBatchNormGrad
FusedBatchNormGrad( const ::tensorflow::Scope & scope, ::tensorflow::Input y_backprop, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input reserve_space_1, ::tensorflow::Input reserve_space_2, const FusedBatchNormGrad::Attrs & attrs )
fonctions statiques publiques
Format de données
Attrs DataFormat( StringPiece x )
Epsilon
Attrs Epsilon( float x )
Formation
Attrs IsTraining( bool x )