Applique des « mises à jour » éparses à des valeurs ou tranches individuelles dans un
variable selon des `indices`.
`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 souhaitons mettre à jour 4 éléments dispersés vers un tenseur de rang 1 à 8 éléments. En Python, cette mise à jour ressemblerait à ceci : ref = tf.Variable([1, 2, 3, 4, 5, 6, 7, 8])
indices = tf.constant([[4], [3], [1] ,[7]])
updates = tf.constant([9, 10, 11, 12])
update = tf.scatter_nd_update(ref, indices, updates)
with tf.Session() as sess:
print sess.run(update)
La mise à jour résultante de la référence ressemblerait à ceci :[1, 11, 3, 10, 9, 6, 7, 12]
Voir tf.scatter_nd
pour plus de détails sur la façon de mettre à jour les tranches.
Classes imbriquées
classe | ResourceScatterNdUpdate.Options | Attributs facultatifs pour ResourceScatterNdUpdate |
Méthodes publiques
statique <T étend le nombre, U> ResourceScatterNdUpdate | |
statique ResourceScatterNdUpdate.Options | useLocking (booléen useLocking) |
Méthodes héritées
Méthodes publiques
public static ResourceScatterNdUpdate 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 ResourceScatterNdUpdate.
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 mises à jour à ajouter à la réf. |
choix | porte des valeurs d'attributs facultatifs |
Retours
- une nouvelle instance de ResourceScatterNdUpdate
public statique ResourceScatterNdUpdate.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. |
---|