classe final pública ResourceApplyKerasMomentum
Atualize '* var' de acordo com o esquema de momentum.
Defina use_nesterov = True se quiser usar o momentum de Nesterov.
acum = acum * momentum - lr * grad var + = acum
Classes aninhadas
aula | ResourceApplyKerasMomentum.Options | Atributos opcionais para ResourceApplyKerasMomentum |
Métodos Públicos
static <T> ResourceApplyKerasMomentum | |
static ResourceApplyKerasMomentum.Options | useLocking (Boolean useLocking) |
static ResourceApplyKerasMomentum.Options | useNesterov (booleano useNesterov) |
Métodos herdados
Métodos Públicos
public static ResourceApplyKerasMomentum create ( escopo de escopo, operando <?> var, operando <?> acum, operando <T> lr, operando <T> grad, operando <T> momentum, opções ...
Método de fábrica para criar uma classe que envolve uma nova operação ResourceApplyKerasMomentum.
Parâmetros
alcance | escopo atual |
---|---|
var | Deve ser de uma variável (). |
acum | Deve ser de uma variável (). |
lr | Fator de escala. Deve ser um escalar. |
grad | O gradiente. |
impulso | Momentum. Deve ser um escalar. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de ResourceApplyKerasMomentum
public static ResourceApplyKerasMomentum.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 ResourceApplyKerasMomentum.Options useNesterov (booleano 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. |
---|