Actualice las entradas relevantes en '* var' y '* acum' de acuerdo con el esquema de impulso.
Establezca use_nesterov = True si desea utilizar el impulso de Nesterov.
Es decir, para las filas para las que hemos graduado, actualizamos var y acumulamos de la siguiente manera:
acumula = acumula * impulso - lr * grad var + = acumula
Clases anidadas
clase | ResourceSparseApplyKerasMomentum.Options | Los atributos opcionales para ResourceSparseApplyKerasMomentum |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
estática <T se extiende Ttype > ResourceSparseApplyKerasMomentum | |
estáticas ResourceSparseApplyKerasMomentum.Options | useLocking (Boolean useLocking) |
estáticas ResourceSparseApplyKerasMomentum.Options | useNesterov (Boolean useNesterov) |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
public static ResourceSparseApplyKerasMomentum crear ( Alcance alcance, operando <?> var, operando <?> acum, operando <T> lr, operando <T> graduado, operando <? extiende TNumber > índices, operando <T> impulso, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación ResourceSparseApplyKerasMomentum.
Parámetros
alcance | alcance actual |
---|---|
var | Debe ser de una variable (). |
acumular | Debe ser de una variable (). |
lr | Tasa de aprendizaje. Debe ser un escalar. |
graduado | El gradiente. |
índices | Un vector de índices en la primera dimensión de var y acum. |
impulso | Impulso. Debe ser un escalar. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de ResourceSparseApplyKerasMomentum
estáticas pública ResourceSparseApplyKerasMomentum.Options useLocking (booleano useLocking)
Parámetros
useLocking | Si es "Verdadero", la actualización de los tensores var y acumuladores estará protegida por un bloqueo; de lo contrario, el comportamiento no está definido, pero puede presentar menos contención. |
---|
public static ResourceSparseApplyKerasMomentum.Options useNesterov (booleano useNesterov)
Parámetros
useNesterov | Si es "Verdadero", el tensor pasado para calcular grad será var + momentum * acum, por lo que al final, el var que obtienes es en realidad var + momentum * acum. |
---|