Cette page a été traduite par l'API Cloud Translation.
Switch to English

tensorflow :: ops :: SparseApplyMomentum

#include <training_ops.h>

Mettre à jour les entrées pertinentes dans « * var » et « * accum » selon le schéma de mouvement.

Résumé

Set use_nesterov = True si vous voulez utiliser l'élan Nesterov.

C'est pour les lignes que nous avons pour GRAD, nous mettons à jour var et accum comme suit:

$$ = accum accum * momentum + grad $$
$$ var - = lr * accum $$

Arguments:

  • portée: A Portée objet
  • var: Devrait être d'une variable ().
  • accum: Devrait être d'une variable ().
  • lr: Taux d'apprentissage. Doit être un scalaire.
  • grad: Le gradient.
  • indices: Un vecteur d'indices dans la première dimension de var et accum.
  • dynamique: Momentum. Doit être un scalaire.

Les attributs facultatifs (voir Attrs ):

  • use_locking: Si True , la mise à jour des tenseurs var et ACCUM seront protégées par une serrure; sinon le comportement est indéfini, mais peut présenter moins affirmation.
  • use_nesterov: Si True , le tenseur est passé à grad Compute sera var - lr * momentum * accum, donc à la fin, le var vous obtenez est en réalité var - lr * momentum * accum.

Retour:

  • Output : Identique à « var ».

Constructeurs et Destructeurs

SparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum)
SparseApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const SparseApplyMomentum::Attrs & attrs)

attributs publics

operation
out

Les fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

fonctions statiques publiques

UseLocking (bool x)
UseNesterov (bool x)

struct

tensorflow :: :: ops SparseApplyMomentum :: ATTRS

Setters d'attributs en option pour SparseApplyMomentum .

attributs publics

opération

 Operation operation

en dehors

:: tensorflow::Output out

Les fonctions publiques

SparseApplyMomentum

 SparseApplyMomentum(
  const :: tensorflow::Scope & scope,
  :: tensorflow::Input var,
  :: tensorflow::Input accum,
  :: tensorflow::Input lr,
  :: tensorflow::Input grad,
  :: tensorflow::Input indices,
  :: tensorflow::Input momentum
)

SparseApplyMomentum

 SparseApplyMomentum(
  const :: tensorflow::Scope & scope,
  :: tensorflow::Input var,
  :: tensorflow::Input accum,
  :: tensorflow::Input lr,
  :: tensorflow::Input grad,
  :: tensorflow::Input indices,
  :: tensorflow::Input momentum,
  const SparseApplyMomentum::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

UseLocking

 Attrs UseLocking(
  bool x
)

UseNesterov

 Attrs UseNesterov(
  bool x
)