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 :

  • Output x_backprop : Un tenseur 4D pour le gradient par rapport à x.
  • scale_backprop Output : un tenseur 1D pour le gradient par rapport à l’échelle.
  • Output offset_backprop : Un tenseur 1D pour le gradient par rapport au décalage.
  • Output reserve_space_3 : Espace réservé inutilisé pour correspondre à l’entrée moyenne dans FusedBatchNorm .
  • Output reserve_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
)