Atualize as entradas relevantes em '* var' e '* acum' de acordo com o esquema de momentum.
Defina use_nesterov = True se quiser usar o momentum de Nesterov.
Isso é para linhas para as quais temos grad, atualizamos var e acumulamos da seguinte forma:
acum = acum * momentum - lr * grad var + = acum
Classes aninhadas
classe | ResourceSparseApplyKerasMomentum.Options | Atributos opcionais para ResourceSparseApplyKerasMomentum |
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
estática <T estende TType > ResourceSparseApplyKerasMomentum | |
estáticos ResourceSparseApplyKerasMomentum.Options | useLocking (booleano useLocking) |
estáticos ResourceSparseApplyKerasMomentum.Options | useNesterov (booleano useNesterov) |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
public static ResourceSparseApplyKerasMomentum criar ( Scope escopo, Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <? estende TNumber > índices, Operando <T> impulso, Opções ... opções)
Método de fábrica para criar uma classe que envolve uma nova operação ResourceSparseApplyKerasMomentum.
Parâmetros
alcance | escopo atual |
---|---|
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 acum. |
impulso | Momentum. Deve ser um escalar. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de ResourceSparseApplyKerasMomentum
public static ResourceSparseApplyKerasMomentum.Options useLocking (Boolean useLocking)
Parâmetros
useLocking | Se for `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. |
---|
public static ResourceSparseApplyKerasMomentum.Options useNesterov (Boolean useNesterov)
Parâmetros
useNesterov | Se for `True`, o tensor passado para calcular grad será var + momentum * acum, então, no final, o var que você obtém é na verdade var + momentum * acum. |
---|