ResourceApplyMomentum

classe final pública ResourceApplyMomentum

Atualize '*var' de acordo com o esquema de momentum.

Defina use_nesterov = True se quiser usar o impulso Nesterov.

acumular = acumular * impulso + grad var -= lr * acumular

Classes aninhadas

aula ResourceApplyMomentum.Options Atributos opcionais para ResourceApplyMomentum

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 > ResourceApplyMomentum
create ( Escopo , Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <T> momentum, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação ResourceApplyMomentum.
ResourceApplyMomentum.Options estático
useLocking (booleano useLocking)
ResourceApplyMomentum.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

Valor constante: "ResourceApplyMomentum"

Métodos Públicos

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

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

Parâmetros
escopo escopo atual
var Deve ser de uma variável().
acumular Deve ser de uma variável().
lr Fator de escala. Deve ser um escalar.
graduado O gradiente.
impulso Momento. Deve ser um escalar.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de ResourceApplyMomentum

public static ResourceApplyMomentum.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 ResourceApplyMomentum.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.