TensorArrayGradWithShape

classe finale publique TensorArrayGradWithShape

Crée un TensorArray pour stocker plusieurs dégradés de valeurs dans le handle donné.

Semblable à TensorArrayGradV3. Cependant, il crée un accumulateur avec une forme développée par rapport au TensorArray d'entrée dont le gradient est calculé. Cela permet de calculer plusieurs gradients pour le même TensorArray à l'aide du même accumulateur.

Méthodes publiques

TensorArrayGradWithShape statique
créer ( Scope scope, Operand <?> handle, Operand <Float> flowIn, Operand <Integer> shapeToPrepend, String source)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TensorArrayGradWithShape.
Sortie <Flottant>
écouler ()
Sortie <?>
gradHandle ()

Méthodes héritées

Méthodes publiques

public static TensorArrayGradWithShape create ( Scope scope, Operand <?> handle, Operand <Float> flowIn, Operand <Integer> shapeToPrepend, String source)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération TensorArrayGradWithShape.

Paramètres
portée portée actuelle
poignée Le handle du TensorArray avant.
fluxIn Un scalaire flottant qui applique un chaînage approprié des opérations.
shapeToPrepend Un vecteur int32 représentant une forme. Les éléments de l'accumulateur de dégradé auront une forme qui est cette valeur shape_to_prepend concaténée avec la forme des éléments du TensorArray correspondant au handle d'entrée.
source La chaîne source du dégradé, utilisée pour décider quel dégradé TensorArray renvoyer.
Retour
  • une nouvelle instance de TensorArrayGradWithShape

sortie publique <Float> flowOut ()

Sortie publique <?> gradHandle ()