flux tensoriel : : opérations : : DispersionMax
#include <state_ops.h>
Réduit les mises à jour éparses en une référence de variable à l'aide de l'opération max
.
Résumé
Cette opération calcule
# Scalar indices ref[indices, ...] = max(ref[indices, ...], updates[...]) # Vector indices (for each i) ref[indices[i], ...] = max(ref[indices[i], ...], updates[i, ...]) # High rank indices (for each i, ..., j) ref[indices[i, ..., j], ...] = max(ref[indices[i, ..., j], ...], updates[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 traité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 = []
.
Arguments :
- scope : un objet Scope
- ref : 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
.
Attributs facultatifs (voir Attrs
) :
- use_locking : 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.
Retours :
-
Output
: = Identique àref
. Renvoyé pour plus de commodité pour les opérations qui souhaitent utiliser les valeurs mises à jour une fois la mise à jour terminée.
Constructeurs et Destructeurs | |
---|---|
ScatterMax (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ScatterMax (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterMax::Attrs & attrs) |
Attributs publics | |
---|---|
operation | |
output_ref |
Fonctions publiques | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fonctions statiques publiques | |
---|---|
UseLocking (bool x) |
Structures | |
---|---|
tensorflow :: ops :: ScatterMax :: Attrs | Setters d'attributs facultatifs pour ScatterMax . |
Attributs publics
opération
Operation operation
réf_sortie
::tensorflow::Output output_ref
Fonctions publiques
DispersionMax
ScatterMax( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
DispersionMax
ScatterMax( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterMax::Attrs & attrs )
nœud
::tensorflow::Node * node() const
opérateur :: tensorflow :: Entrée
operator::tensorflow::Input() const
opérateur :: tensorflow :: Sortie
operator::tensorflow::Output() const
Fonctions statiques publiques
UtiliserVerrouillage
Attrs UseLocking( bool x )