flux tensoriel : : opérations : : Accumulateur conditionnel

#include <data_flow_ops.h>

Un accumulateur conditionnel pour agréger les gradients.

Résumé

L'accumulateur accepte les gradients marqués d'un local_step supérieur ou égal au global_step le plus récent connu de l'accumulateur. La moyenne peut être extraite de l'accumulateur, à condition que suffisamment de gradients aient été accumulés. L'extraction de la moyenne réinitialise automatiquement l'agrégat à 0 et incrémente le global_step enregistré par l'accumulateur.

Arguments:

  • scope : un objet Scope
  • dtype : Le type de la valeur accumulée.
  • shape : La forme des valeurs peut être [], auquel cas la forme est inconnue.

Attributs facultatifs (voir Attrs ) :

  • conteneur : S'il n'est pas vide, cet accumulateur est placé dans le conteneur donné. Sinon, un conteneur par défaut est utilisé.
  • shared_name : s'il n'est pas vide, cet accumulateur sera partagé sous le nom donné sur plusieurs sessions.

Retour:

  • Output : la poignée de l'accumulateur.

Constructeurs et Destructeurs

ConditionalAccumulator (const :: tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape)
ConditionalAccumulator (const :: tensorflow::Scope & scope, DataType dtype, PartialTensorShape shape, const ConditionalAccumulator::Attrs & attrs)

Attributs publics

handle
operation

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fonctions statiques publiques

Container (StringPiece x)
ReductionType (StringPiece x)
SharedName (StringPiece x)

Structures

tensorflow :: ops :: ConditionalAccumulator :: Attrs

Setters d'attributs facultatifs pour ConditionalAccumulator .

Attributs publics

poignée

::tensorflow::Output handle

opération

Operation operation

Fonctions publiques

Accumulateur conditionnel

 ConditionalAccumulator(
  const ::tensorflow::Scope & scope,
  DataType dtype,
  PartialTensorShape shape
)

Accumulateur conditionnel

 ConditionalAccumulator(
  const ::tensorflow::Scope & scope,
  DataType dtype,
  PartialTensorShape shape,
  const ConditionalAccumulator::Attrs & attrs
)

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: Entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const 

Fonctions statiques publiques

Récipient

Attrs Container(
  StringPiece x
)

Type de réduction

Attrs ReductionType(
  StringPiece x
)

NomPartagé

Attrs SharedName(
  StringPiece x
)