ResourceSparseApplyKerasMomentum

ResourceSparseApplyKerasMomentum classe final pública

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 acumulando da seguinte forma:

acum = acum * momentum - lr * grad var + = acum

Classes aninhadas

classe ResourceSparseApplyKerasMomentum.Options Atributos opcionais para ResourceSparseApplyKerasMomentum

Métodos Públicos

estática <T, U estende Number> ResourceSparseApplyKerasMomentum
criar ( Scope escopo, Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <U> índices, Operando <T> impulso, Opções ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação ResourceSparseApplyKerasMomentum.
estáticos ResourceSparseApplyKerasMomentum.Options
useLocking (booleano useLocking)
estáticos ResourceSparseApplyKerasMomentum.Options
useNesterov (booleano useNesterov)

Métodos herdados

Métodos Públicos

public static ResourceSparseApplyKerasMomentum criar ( Scope escopo, Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <U> í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.