Un accumulateur conditionnel pour agréger les gradients.
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. Il s'agit d'une version ressource de ConditionalAccumulator qui fonctionnera dans TF2.0 avec tf.cond version 2.
Classes imbriquées
classe | ResourceConditionalAccumulator.Options | Attributs facultatifs pour ResourceConditionalAccumulator |
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
Sortie < TType > | comme Sortie () Renvoie le handle symbolique du tenseur. |
statique ResourceConditionalAccumulator.Options | conteneur (conteneur de chaîne) |
statique <T étend TType > ResourceConditionalAccumulator | créer (portée de portée , type de classe<T>, forme de forme , options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération ResourceConditionalAccumulator. |
Sortie <?> | poignée () La poignée de l'accumulateur. |
statique ResourceConditionalAccumulator.Options | réductionType (Chaîne réductionType) |
statique ResourceConditionalAccumulator.Options | nom partagé (chaîne nom partagé) |
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
sortie publique < TType > asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
Conteneur public statique ResourceConditionalAccumulator.Options (conteneur String)
Paramètres
récipient | S'il n'est pas vide, cet accumulateur est placé dans le conteneur donné. Sinon, un conteneur par défaut est utilisé. |
---|
public static ResourceConditionalAccumulator créer ( Scope scope, Class<T> dtype, Shape shape, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ResourceConditionalAccumulator.
Paramètres
portée | portée actuelle |
---|---|
type | Type de valeur accumulée. |
forme | La forme des valeurs peut être [], auquel cas la forme est inconnue. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de ResourceConditionalAccumulator
public statique ResourceConditionalAccumulator.Options sharedName (String sharedName)
Paramètres
Nom partagé | S'il n'est pas vide, cet accumulateur sera partagé sous le nom donné sur plusieurs sessions. |
---|