classe final pública ResourceApplyKerasMomentum
Atualize '*var' de acordo com o esquema de momento.
Defina use_nesterov = True se você quiser usar o momento Nesterov.
acumulo = acumulo * momento - lr * grad var += acumulo
Classes aninhadas
| aula | ResourceApplyKerasMomentum.Options | Atributos opcionais para ResourceApplyKerasMomentum | |
Métodos públicos
| estático <T> ResourceApplyKerasMomentum | |
| static ResourceApplyKerasMomentum.Options | useLocking (booleano useLocking) |
| static ResourceApplyKerasMomentum.Options | useNesterov (booleano useNesterov) |
Métodos Herdados
Métodos públicos
public static ResourceApplyKerasMomentum create ( Escopo do 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 envolve uma nova operação ResourceApplyKerasMomentum.
Parâmetros
| alcance | escopo atual |
|---|---|
| var | Deve ser de uma variável(). |
| acumular | Deve ser de uma variável(). |
| lr | Fator de escala. Deve ser um escalar. |
| graduar | O gradiente. |
| impulso | Impulso. 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
| useBloqueio | Se `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 (Boolean useNesterov)
Parâmetros
| useNesterov | 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. |
|---|