Actualice las entradas relevantes en '*var' y '*accum' según el esquema de impulso.
Establezca use_nesterov = True si desea utilizar el impulso de Nesterov.
Es decir, para las filas para las que tenemos grad, actualizamos var y accum de la siguiente manera:
acumulación = acumulación * impulso - lr * grad var += acumulación
Clases anidadas
clase | ResourceSparseApplyKerasMomentum.Options | Atributos opcionales para ResourceSparseApplyKerasMomentum |
Constantes
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
estático <T extiende TType > ResourceSparseApplyKerasMomentum | |
estático ResourceSparseApplyKerasMomentum.Options | useLocking (uso booleano Locking) |
estático ResourceSparseApplyKerasMomentum.Options | useNesterov (uso booleanoNesterov) |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
public static ResourceSparseApplyKerasMomentum create (Ámbito de alcance , Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <? extiende TNumber > índices, Operando <T> impulso, Opciones... opciones)
Método de fábrica para crear una clase que envuelve 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 accum. |
impulso | Impulso. Debe ser un escalar. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de ResourceSparseApplyKerasMomentum
ResourceSparseApplyKerasMomentum.Options estático público useLocking (useLocking booleano)
Parámetros
utilizarBloqueo | Si es "True", la actualización de los tensores var y accum estará protegida por un bloqueo; de lo contrario, el comportamiento no está definido, pero puede presentar menos contención. |
---|
público estático ResourceSparseApplyKerasMomentum.Options useNesterov (uso booleanoNesterov)
Parámetros
utilizarNesterov | Si es "Verdadero", el tensor pasado para calcular grad será var + impulso * accum, por lo que al final, el var que obtienes es en realidad var + impulso * accum. |
---|