Atualize as entradas relevantes em '*var' e '*accum' de acordo com o esquema de impulso.
Defina use_nesterov = True se quiser usar o impulso Nesterov.
Ou seja, para as linhas para as quais graduamos, atualizamos var e accum da seguinte forma:
acumular = acumular * impulso + grad var -= lr * acumular
Classes aninhadas
aula | ResourceSparseApplyMomentum.Options | Atributos opcionais para ResourceSparseApplyMomentum |
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
estático <T estende TType > ResourceSparseApplyMomentum | |
ResourceSparseApplyMomentum.Options estático | useLocking (booleano useLocking) |
ResourceSparseApplyMomentum.Options estático | useNesterov (booleano useNesterov) |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
public static ResourceSparseApplyMomentum create ( Escopo escopo , Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <? estende TNumber > índices, Operando <T> momentum, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação ResourceSparseApplyMomentum.
Parâmetros
escopo | escopo atual |
---|---|
var | Deve ser de uma variável(). |
acumular | Deve ser de uma variável(). |
lr | Taxa de Aprendizagem. Deve ser um escalar. |
graduado | O gradiente. |
índices | Um vetor de índices na primeira dimensão de var e accum. |
impulso | Momento. Deve ser um escalar. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de ResourceSparseApplyMomentum
public static ResourceSparseApplyMomentum.Options useLocking (booleano useLocking)
Parâmetros
useLocking | Se `True`, a atualização dos tensores var e accum será protegida por um bloqueio; caso contrário, o comportamento será indefinido, mas poderá apresentar menos contenção. |
---|
public static ResourceSparseApplyMomentum.Options useNesterov (booleano useNesterov)
Parâmetros
usarNesterov | Se `True`, o tensor passado para calcular grad será var - lr * momentum * accum, então, no final, o var que você obtém é na verdade var - lr * momentum * accum. |
---|