flux tensoriel : : opérations : : DispersionAjouter

#include <state_ops.h>

Ajoute des mises à jour éparses à une référence de variable.

Résumé

Cette opération calcule

# Scalar indices
ref[indices, ...] += updates[...]

# Vector indices (for each i)
ref[indices[i], ...] += updates[i, ...]

# High rank indices (for each i, ..., j)
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 gérées correctement : si plusieurs indices font référence au même emplacement, leurs contributions s'ajoutent.

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 à ajouter à ref .

Attributs facultatifs (voir Attrs ) :

  • use_locking : Si True, l'ajout sera protégé par un verrou ; sinon, le comportement n'est pas défini, mais peut présenter moins de conflits.

Retours :

  • Output : = Identique à ref . Renvoyé pour faciliter les opérations qui souhaitent utiliser les valeurs mises à jour une fois la mise à jour terminée.

Constructeurs et Destructeurs

ScatterAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates)
ScatterAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterAdd::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 :: ScatterAdd :: Attrs

Setters d'attributs facultatifs pour ScatterAdd .

Attributs publics

opération

Operation operation

réf_sortie

::tensorflow::Output output_ref

Fonctions publiques

DispersionAjouter

 ScatterAdd(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input ref,
  ::tensorflow::Input indices,
  ::tensorflow::Input updates
)

DispersionAjouter

 ScatterAdd(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input ref,
  ::tensorflow::Input indices,
  ::tensorflow::Input updates,
  const ScatterAdd::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
)