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 | |
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