flux tensoriel : : opérations : : ResourceSparseApplyMomentum : Attributs

#include <training_ops.h>

Setters d'attributs facultatifs pour ResourceSparseApplyMomentum .

Résumé

Attributs publics

use_locking_ = false
bool
use_nesterov_ = false
bool

Fonctions publiques

UseLocking (bool x)
TF_MUST_USE_RESULT Attrs
Si True , la mise à jour des tenseurs var et cumul sera protégée par un verrou ; sinon, le comportement n'est pas défini, mais peut présenter moins de conflits.
UseNesterov (bool x)
TF_MUST_USE_RESULT Attrs
Si True , le tenseur passé au calcul de grad sera var - lr * momentum * accum, donc à la fin, le var que vous obtenez est en fait var - lr * momentum * accum.

Attributs publics

utiliser_verrouillage_

bool tensorflow::ops::ResourceSparseApplyMomentum::Attrs::use_locking_ = false

utiliser_nesterov_

bool tensorflow::ops::ResourceSparseApplyMomentum::Attrs::use_nesterov_ = false

Fonctions publiques

UtiliserVerrouillage

TF_MUST_USE_RESULT Attrs tensorflow::ops::ResourceSparseApplyMomentum::Attrs::UseLocking(
  bool x
)

Si True , la mise à jour des tenseurs var et cumul sera protégée par un verrou ; sinon, le comportement n'est pas défini, mais peut présenter moins de conflits.

La valeur par défaut est false

Utiliser Nesterov

TF_MUST_USE_RESULT Attrs tensorflow::ops::ResourceSparseApplyMomentum::Attrs::UseNesterov(
  bool x
)

Si True , le tenseur passé au calcul de grad sera var - lr * momentum * accum, donc à la fin, le var que vous obtenez est en fait var - lr * momentum * accum.

La valeur par défaut est false