Réduit les mises à jour éparses en une référence de variable à l'aide de l'opération « max ».
Cette opération calcule
# Indices scalaires ref[indices, ...] = max(ref[indices, ...], mises à jour[...])
# Indices vectoriels (pour chaque i) ref[indices[i], ...] = max(ref[indices[i], ...], mises à jour[i, ...])
# Indices de rang élevé (pour chaque i, ..., j) ref[indices[i, ..., j], ...] = max(ref[indices[i, ..., j], .. .], mises à jour[i, ..., j, ...])
Cette opération génère `ref` une fois la mise à jour terminée. Cela facilite l’enchaînement des opérations nécessitant l’utilisation de la valeur de réinitialisation.
Les entrées en double sont gérées correctement : si plusieurs « indices » font référence au même emplacement, leurs contributions se combinent.
Nécessite `updates.shape = indices.shape + ref.shape[1:]` ou `updates.shape = []`.
Classes imbriquées
classe | DispersionMax.Options | Attributs facultatifs pour ScatterMax |
Méthodes publiques
Sortie <T> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
statique <T étend le nombre, U étend le nombre> ScatterMax <T> | |
Sortie <T> | sortieRef () = Identique à `ref`. |
ScatterMax.Options statiques | useLocking (booléen useLocking) |
Méthodes héritées
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique d'un 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.
public static ScatterMax <T> créer (portée de portée , référence d'opérande <T>, indices d'opérande <U>, mises à jour d'opérande <T>, options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ScatterMax.
Paramètres
portée | portée actuelle |
---|---|
réf | Doit provenir d'un nœud « Variable ». |
indices | Un tenseur d'indices dans la première dimension de `ref`. |
mises à jour | Un tenseur de valeurs mises à jour à réduire en `ref`. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de ScatterMax
sortie publique <T> sortieRef ()
= Identique à `ref`. Renvoyé pour faciliter les opérations qui souhaitent utiliser les valeurs mises à jour une fois la mise à jour terminée.
public statique ScatterMax.Options useLocking (booléen useLocking)
Paramètres
utiliserVerrouillage | Si True, la mise à jour sera protégée par un verrou ; sinon, le comportement n'est pas défini, mais peut présenter moins de conflits. |
---|