Merken Sie den Termin vor! Google I / O kehrt vom 18. bis 20. Mai zurück Registrieren Sie sich jetzt
Diese Seite wurde von der Cloud Translation API übersetzt.
Switch to English

Tensorflow :: ops :: ResourceSparseApplyRMSProp

#include <training_ops.h>

Aktualisieren Sie '* var' gemäß dem RMSProp-Algorithmus.

Zusammenfassung

Beachten Sie, dass bei einer dichten Implementierung dieses Algorithmus ms und mom auch dann aktualisiert werden, wenn der Grad Null ist. Bei dieser spärlichen Implementierung werden ms und mom jedoch nicht in Iterationen aktualisiert, bei denen der Grad Null ist.

mean_square = Zerfall * mean_square + (1-Zerfall) * Gradient ** 2 Delta = Lernrate * Gradient / Quadrat (mean_square + epsilon)

ms <- rho * ms_ {t-1} + (1-rho) * grad * grad mom <- Impuls * mom_ {t-1} + lr * grad / sqrt (ms + epsilon) var <- var - mom

Argumente:

  • scope: Ein Scope- Objekt
  • var: Sollte von einer Variablen stammen ().
  • ms: Sollte von einer Variablen stammen ().
  • Mutter: Sollte von einer Variablen stammen ().
  • lr: Skalierungsfaktor. Muss ein Skalar sein.
  • Rho: Zerfallsrate. Muss ein Skalar sein.
  • epsilon: Ridge Begriff. Muss ein Skalar sein.
  • grad: Der Gradient.
  • Indizes: Ein Vektor von Indizes in die erste Dimension von var, ms und mom.

Optionale Attribute (siehe Attrs ):

  • use_locking: Wenn True , wird die Aktualisierung der var-, ms- und mom-Tensoren durch eine Sperre geschützt. Andernfalls ist das Verhalten undefiniert, weist jedoch möglicherweise weniger Konflikte auf.

Kehrt zurück:

Konstruktoren und Destruktoren

ResourceSparseApplyRMSProp (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input ms, :: tensorflow::Input mom, :: tensorflow::Input lr, :: tensorflow::Input rho, :: tensorflow::Input momentum, :: tensorflow::Input epsilon, :: tensorflow::Input grad, :: tensorflow::Input indices)
ResourceSparseApplyRMSProp (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input ms, :: tensorflow::Input mom, :: tensorflow::Input lr, :: tensorflow::Input rho, :: tensorflow::Input momentum, :: tensorflow::Input epsilon, :: tensorflow::Input grad, :: tensorflow::Input indices, const ResourceSparseApplyRMSProp::Attrs & attrs)

Öffentliche Attribute

operation

Öffentliche Funktionen

operator::tensorflow::Operation () const

Öffentliche statische Funktionen

UseLocking (bool x)

Strukturen

tensorflow :: ops :: ResourceSparseApplyRMSProp :: Attrs

Optionale Attributsetzer für ResourceSparseApplyRMSProp .

Öffentliche Attribute

Operation

Operation operation

Öffentliche Funktionen

ResourceSparseApplyRMSProp

 ResourceSparseApplyRMSProp(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input ms,
  ::tensorflow::Input mom,
  ::tensorflow::Input lr,
  ::tensorflow::Input rho,
  ::tensorflow::Input momentum,
  ::tensorflow::Input epsilon,
  ::tensorflow::Input grad,
  ::tensorflow::Input indices
)

ResourceSparseApplyRMSProp

 ResourceSparseApplyRMSProp(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input var,
  ::tensorflow::Input ms,
  ::tensorflow::Input mom,
  ::tensorflow::Input lr,
  ::tensorflow::Input rho,
  ::tensorflow::Input momentum,
  ::tensorflow::Input epsilon,
  ::tensorflow::Input grad,
  ::tensorflow::Input indices,
  const ResourceSparseApplyRMSProp::Attrs & attrs
)

operator :: tensorflow :: Operation

 operator::tensorflow::Operation() const 

Öffentliche statische Funktionen

UseLocking

Attrs UseLocking(
  bool x
)