tensorflow :: operaciones :: ResourceApplyMomentum
#include <training_ops.h>Actualice '* var' de acuerdo con el esquema de impulso.
Resumen
Establezca use_nesterov = True si
quiere utilizar el impulso de Nesterov.
acumula = acumula * impulso + grad var - = lr * acumula
Argumentos:
- alcance: un objeto de alcance
- var: debe ser de una variable ().
- acum: debe ser de una variable ().
- lr: factor de escala. Debe ser un escalar.
- grad: El gradiente.
- impulso: impulso. Debe ser un escalar.
Atributos opcionales (consulte Attrs ):
- use_locking: Si es
True, la actualización de los tensores var y acumuladores estará protegida por un bloqueo; de lo contrario, el comportamiento no está definido, pero puede mostrar menos contención. - use_nesterov: Si es
True, el tensor pasado para calcular grad será var - lr * momentum * acum, por lo que al final, la var que obtienes es en realidad var - lr * momentum * acum.
Devoluciones:
- la
Operationcreada
Constructores y Destructores | |
|---|---|
ResourceApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum) | |
ResourceApplyMomentum (const :: tensorflow::Scope & scope, :: tensorflow::Input var, :: tensorflow::Input accum, :: tensorflow::Input lr, :: tensorflow::Input grad, :: tensorflow::Input momentum, const ResourceApplyMomentum::Attrs & attrs) |
Atributos públicos | |
|---|---|
operation | |
Funciones publicas | |
|---|---|
operator::tensorflow::Operation () const | |
Funciones estáticas públicas | |
|---|---|
UseLocking (bool x) | |
UseNesterov (bool x) | |
Estructuras | |
|---|---|
| tensorflow :: ops :: ResourceApplyMomentum :: Attrs | Establecedores de atributos opcionales para ResourceApplyMomentum . |
Atributos públicos
operación
Operation operation
Funciones publicas
ResourceApplyMomentum
ResourceApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum )
ResourceApplyMomentum
ResourceApplyMomentum( const ::tensorflow::Scope & scope, ::tensorflow::Input var, ::tensorflow::Input accum, ::tensorflow::Input lr, ::tensorflow::Input grad, ::tensorflow::Input momentum, const ResourceApplyMomentum::Attrs & attrs )
operador :: tensorflow :: Operación
operator::tensorflow::Operation() const
Funciones estáticas públicas
UseLocking
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )