flux tensoriel : : opérations : : ResourceScatterNdAdd
#include <state_ops.h>
Applique une addition clairsemée à des valeurs ou tranches individuelles dans une Variable .
Résumé
ref
est un Tensor
de rang P
et indices
est un Tensor
de rang Q
.
indices
doivent être des tenseurs entiers, contenant des indices dans ref
. Il doit avoir la forme [d_0, ..., d_{Q-2}, K]
où 0 < K <= P
.
La dimension la plus intérieure des indices
(de longueur K
) correspond aux indices en éléments (si K = P
) ou en tranches (si K < P
) le long de la K
ème dimension de ref
.
updates
sont Tensor
de rang Q-1+PK
de forme :
[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]
Par exemple, disons que nous voulons ajouter 4 éléments dispersés à un tenseur de rang 1 à 8 éléments. En Python, cet ajout ressemblerait à ceci :
ref = tf.Variable([1, 2, 3, 4, 5, 6, 7, 8], use_resource=True) indices = tf.constant([[4], [3], [1], [7]]) updates = tf.constant([9, 10, 11, 12]) add = tf.scatter_nd_add(ref, indices, updates) with tf.Session() as sess: print sess.run(add)
La mise à jour résultante de la référence ressemblerait à ceci :
[1, 13, 3, 14, 14, 6, 7, 20]
Voir tf.scatter_nd
pour plus de détails sur la façon de mettre à jour les tranches.
Arguments :
- scope : un objet Scope
- ref : un descripteur de ressource. Doit provenir d'un VarHandleOp.
- indices : un tenseur . Doit être l'un des types suivants : int32, int64. Un tenseur d'indices dans la réf.
- mises à jour : Un Tenseur . Doit être du même type que la réf. Un tenseur de valeurs à ajouter à la réf.
Attributs facultatifs (voir Attrs
) :
- use_locking : un booléen facultatif. La valeur par défaut est True. Si True, l'affectation sera protégée par un verrou ; sinon, le comportement n'est pas défini, mais peut présenter moins de conflits.
Retours :
- l'
Operation
créée
Constructeurs et Destructeurs | |
---|---|
ResourceScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ResourceScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ResourceScatterNdAdd::Attrs & attrs) |
Attributs publics | |
---|---|
operation |
Fonctions publiques | |
---|---|
operator::tensorflow::Operation () const |
Fonctions statiques publiques | |
---|---|
UseLocking (bool x) |
Structures | |
---|---|
tensorflow :: ops :: ResourceScatterNdAdd :: Attrs | Setters d'attributs facultatifs pour ResourceScatterNdAdd . |
Attributs publics
opération
Operation operation
Fonctions publiques
ResourceScatterNdAdd
ResourceScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ResourceScatterNdAdd
ResourceScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ResourceScatterNdAdd::Attrs & attrs )
opérateur :: tensorflow :: Opération
operator::tensorflow::Operation() const
Fonctions statiques publiques
UtiliserVerrouillage
Attrs UseLocking( bool x )