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 | AplicarMomentum.Options | Atributos opcionais para ApplyMomentum | |
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
| Saída <T> | asOutput () Retorna o identificador simbólico do tensor. |
| estático <T estende TType > ApplyMomentum <T> | |
| Saída <T> | fora () O mesmo que "var". |
| ApplyMomentum.Options estático | useLocking (booleano useLocking) |
| ApplyMomentum.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
Métodos Públicos
Saída pública <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.
public static ApplyMomentum <T> create ( Escopo escopo , Operando <T> var, Operando <T> 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 ApplyMomentum.
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 do ApplyMomentum
public static ApplyMomentum.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 ApplyMomentum.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. |
|---|