Ajoute des opérations pour calculer les dérivées partielles de la somme de y
s par rapport à x
s, c'est-à-dire d(y_1 + y_2 + ...)/dx_1, d(y_1 + y_2 + ...)/dx_2...
Si les valeurs Options.dx()
sont définies, elles constituent les dérivées partielles symboliques initiales d'une fonction de perte L
par rapport à y
. Options.dx()
doit avoir la taille de y
.
Si Options.dx()
n'est pas défini, l'implémentation utilisera dx de OnesLike
pour toutes les formes de y
.
Les dérivées partielles sont renvoyées dans la sortie dy
, avec la taille de x
.
Exemple d'utilisation :
Gradients gradients = Gradients.create(scope, Arrays.asList(loss), Arrays.asList(w, b));
Constant<Float> alpha = ops.constant(1.0f, Float.class);
ApplyGradientDescent.create(scope, w, alpha, gradients.<Float>dy(0));
ApplyGradientDescent.create(scope, b, alpha, gradients.<Float>dy(1));
Classes imbriquées
classe | Dégradés.Options | Attributs facultatifs pour Gradients |
Méthodes publiques
Dégradés statiques | create ( Scope scope, Operand <?> y, Iterable<? extends Operand <?>> x, Options... options) Ajoute des opérations de calcul de gradients au graphique en fonction de la portée. |
Dégradés statiques | créer ( Portée de la portée, Iterable<? étend l'opérande <?>> y, Iterable<? étend l'opérande <?>> x, Options... options) Ajoute des opérations de calcul de gradients au graphique en fonction de la portée. |
Dégradés statiques.Options | dx (Iterable<? étend l'opérande <?>> dx) |
<T> Sortie <T> | dy (index int) Renvoie un handle symbolique vers l'une des sorties de l'opération de dégradé Attention : ne vérifie pas que le type du tenseur correspond à T. |
Liste< Sortie <?>> | dy () Dérivées partielles de y s par rapport à |
Itérateur< Opérande <?>> | itérateur () |
Méthodes héritées
Méthodes publiques
Les dégradés statiques publics créent ( Scope scope, Operand <?> y, Iterable<? extends Operand <?>> x, Options... options)
Ajoute des opérations de calcul de gradients au graphique en fonction de la portée.
Il s'agit d'une version simplifiée de create(Scope, Iterable, Iterable, Options)
où y
est une sortie unique.
Paramètres
portée | portée actuelle du graphique |
---|---|
oui | sortie de la fonction à dériver |
x | entrées de la fonction pour laquelle les dérivées partielles sont calculées |
choix | porte des valeurs d'attributs facultatifs |
Retours
- une nouvelle instance de
Gradients
Jetés
IllegalArgumentException | si l'environnement d'exécution n'est pas un graphique |
---|
Les dégradés statiques publics créent ( Scope scope, Iterable<? extends Operand <?>> y, Iterable<? extends Operand <?>> x, Options... options)
Ajoute des opérations de calcul de gradients au graphique en fonction de la portée.
Paramètres
portée | portée actuelle du graphique |
---|---|
oui | sorties de la fonction à dériver |
x | entrées de la fonction pour laquelle les dérivées partielles sont calculées |
choix | porte des valeurs d'attributs facultatifs |
Retours
- une nouvelle instance de
Gradients
Jetés
IllegalArgumentException | si l'environnement d'exécution n'est pas un graphique |
---|
public static Gradients.Options dx (Iterable<? extends Operand <?>> dx)
Paramètres
dx | dérivées partielles d'une fonction de perte L par rapport à y |
---|
Retours
- constructeur pour ajouter plus d'options à cette opération
sortie publique <T> dy (index int)
Renvoie un handle symbolique vers l'une des sorties de l'opération de dégradé
Attention : ne vérifie pas que le type du tenseur correspond à T. Il est recommandé d'appeler cette méthode avec un paramètre de type explicite plutôt que de le laisser déduire, par exemple gradients.<Float>dy(0)
Paramètres
indice | L'index de la sortie parmi les dégradés ajoutés par cette opération |
---|