Atualize '* var' de acordo com o esquema de momentum.
Defina use_nesterov = True se quiser usar o momentum de Nesterov.
acum = acum * momentum + grad var - = lr * acum
Classes aninhadas
| classe | ApplyMomentum.Options | Atributos opcionais para ApplyMomentum | |
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
| Output <T> | asOutput () Retorna o identificador simbólico do tensor. |
| estática <T estende TType > ApplyMomentum <T> | |
| Output <T> | a () O mesmo que "var". |
| estáticos ApplyMomentum.Options | useLocking (booleano useLocking) |
| estáticos ApplyMomentum.Options | useNesterov (booleano useNesterov) |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para as 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 ( Scope escopo, Operando <T> var, Operando <T> accum, Operando <T> lr, Operando <T> grad, Operando <T> impulso, Opções ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação ApplyMomentum.
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 ApplyMomentum
public static ApplyMomentum.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 ApplyMomentum.Options useNesterov (Boolean useNesterov)
Parâmetros
| useNesterov | Se for `True`, o tensor passado para calcular grad será var - lr * momento * acum, então, no final, o var que você obtém é na verdade var - lr * momento * acum. |
|---|