tensorflow :: ops :: ResourceSparseApplyKerasMomentum
#include <training_ops.h>
Atualize as entradas relevantes em '* var' e '* acum' de acordo com o esquema de momentum.
Resumo
Defina use_nesterov = True se quiser usar o momentum de Nesterov.
Isso é para linhas para as quais temos grad, atualizamos var e acumulando da seguinte forma:
acum = acum * momentum - lr * grad var + = acum
Argumentos:
- escopo: um objeto Scope
- var: deve ser de uma variável ().
- acum: deve ser de uma variável ().
- lr: Taxa de aprendizagem. Deve ser um escalar.
- grad: O gradiente.
- índices: Um vetor de índices na primeira dimensão de var e de acum.
- momentum: Momentum. Deve ser um escalar.
Atributos opcionais (consulte Attrs
):
- use_locking: Se
True
, a atualização dos tensores var e Accum será protegida por um bloqueio; caso contrário, o comportamento é indefinido, mas pode exibir menos contenção. - use_nesterov: Se
True
, o tensor passado para calcular grad será var + momento * acum, então, no final, o var que você obtém é realmente var + momento * acum.
Retorna:
- a
Operation
criada
Construtores e Destruidores | |
---|---|
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) |
Atributos públicos | |
---|---|
operation |
Funções públicas | |
---|---|
operator::tensorflow::Operation () const |
Funções estáticas públicas | |
---|---|
UseLocking (bool x) | |
UseNesterov (bool x) |
Structs | |
---|---|
tensorflow :: ops :: ResourceSparseApplyKerasMomentum :: Attrs | Configuradores de atributos opcionais para ResourceSparseApplyKerasMomentum . |
Atributos públicos
Operação
Operation operation
Funções públicas
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 )
operador :: tensorflow :: Operação
operator::tensorflow::Operation() const
Funções estáticas públicas
UseLocking
Attrs UseLocking( bool x )
UseNesterov
Attrs UseNesterov( bool x )