ResourceSparseApplyKerasMomentum

classe final pública ResourceSparseApplyKerasMomentum

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 - lr * grad var += acumular

Classes aninhadas

aula ResourceSparseApplyKerasMomentum.Options Atributos opcionais para ResourceSparseApplyKerasMomentum

Métodos Públicos

estático <T, U estende o número> ResourceSparseApplyKerasMomentum
create ( Escopo , Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Índices de Operando <U>, Operando <T> momentum, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação ResourceSparseApplyKerasMomentum.
ResourceSparseApplyKerasMomentum.Options estático
useLocking (booleano useLocking)
ResourceSparseApplyKerasMomentum.Options estático
useNesterov (booleano useNesterov)

Métodos herdados

Métodos Públicos

public static ResourceSparseApplyKerasMomentum create ( Escopo escopo , Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <U> índices, Operando <T> momentum, Opções... opções)

Método de fábrica para criar uma classe que agrupa uma nova operação ResourceSparseApplyKerasMomentum.

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 ResourceSparseApplyKerasMomentum

public static ResourceSparseApplyKerasMomentum.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 ResourceSparseApplyKerasMomentum.Options useNesterov (booleano useNesterov)

Parâmetros
usarNesterov Se `True`, o tensor passado para calcular grad será var + momentum * accum, então, no final, o var que você obtém é na verdade var + momentum * accum.