Tensorflow :: ops :: ResourceSparseApplyKerasMomentum

#include <training_ops.h>

Aktualisieren Sie relevante Einträge in '* var' und '* accum' gemäß dem Momentum-Schema.

Zusammenfassung

Setzen Sie use_nesterov = True, wenn Sie den Nesterov-Impuls verwenden möchten.

Das heißt, für Zeilen, für die wir grad haben, aktualisieren wir var und accum wie folgt:

akkumulieren = akkumulieren * Impuls - lr * grad var + = akkumulieren

Argumente:

• scope: Ein Scope- Objekt
• var: Sollte von einer Variablen stammen ().
• accum: Sollte von einer Variablen stammen ().
• lr: Lernrate. Muss ein Skalar sein.
• grad: Der Gradient.
• Indizes: Ein Vektor von Indizes in die erste Dimension von var und accum.
• Momentum: Momentum. Muss ein Skalar sein.

Optionale Attribute (siehe Attrs ):

• use_locking: Wenn True , wird die Aktualisierung der Var- und Accum-Tensoren durch eine Sperre geschützt. Andernfalls ist das Verhalten undefiniert, weist jedoch möglicherweise weniger Konflikte auf.
• use_nesterov: Wenn True , ist der Tensor, der zur Berechnung von grad übergeben wird, var + Impuls * akkumulieren. Am Ende ist die Var, die Sie erhalten, tatsächlich var + Impuls * akkumulieren.

Kehrt zurück:

• die erstellte Operation

Konstruktoren und Destruktoren
ResourceSparseApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum)
ResourceSparseApplyKerasMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input indices, :: tensorflow::Input momentum, const ResourceSparseApplyKerasMomentum::Attrs & attrs)

Öffentliche Attribute
operation	Operation

Öffentliche Funktionen
operator::tensorflow::Operation () const

Öffentliche statische Funktionen
UseLocking (bool x)	Attrs
UseNesterov (bool x)	Attrs

Strukturen
tensorflow :: ops :: ResourceSparseApplyKerasMomentum :: Attrs	Optionale Attributsetzer für ResourceSparseApplyKerasMomentum .

Öffentliche Attribute

Operation

Operation operation

Öffentliche Funktionen

ResourceSparseApplyKerasMomentum

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

ResourceSparseApplyKerasMomentum

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

operator :: tensorflow :: Operation

 operator::tensorflow::Operation() const 

Öffentliche statische Funktionen

UseLocking

Attrs UseLocking(
  bool x
)

Verwenden Sie Nesterov

Attrs UseNesterov(
  bool x
)