Applique une addition clairsemée à des valeurs individuelles ou à des tranches dans une variable.
`ref` est un `Tensor` de rang `P` et `indices` est un `Tensor` de rang `Q`.
`indices` doit être un tenseur entier, 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 » dimension de « ref ».
`updates` est un `Tensor` de rang `Q-1+PK` avec la 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.
Classes imbriquées
classe | ResourceScatterNdAdd.Options | Attributs facultatifs pour ResourceScatterNdAdd |
Méthodes publiques
statique <T étend le nombre, U> ResourceScatterNdAdd | |
statique ResourceScatterNdAdd.Options | useLocking (booléen useLocking) |
Méthodes héritées
Méthodes publiques
public static ResourceScatterNdAdd create ( Scope scope, Operand <?> ref, Operand <T> indices, Operand <U> mises à jour, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération ResourceScatterNdAdd.
Paramètres
portée | portée actuelle |
---|---|
réf | Un handle 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. |
choix | porte des valeurs d'attributs facultatifs |
Retours
- une nouvelle instance de ResourceScatterNdAdd
public statique ResourceScatterNdAdd.Options useLocking (booléen useLocking)
Paramètres
utiliserVerrouillage | 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. |
---|