Actualice las entradas relevantes en '*var' y '*accum' según el esquema de momento.
Establezca use_nesterov = True si desea utilizar el impulso de Nesterov.
Eso es para las filas para las que tenemos graduación, actualizamos var y accum de la siguiente manera:
accum = accum * impulso - lr * grad var += accum
Clases anidadas
clase | ResourceSparseApplyKerasMomentum.Options | Atributos opcionales para ResourceSparseApplyKerasMomentum |
Métodos públicos
estático <T, U extiende Número> ResourceSparseApplyKerasMomentum | |
Estático ResourceSparseApplyKerasMomentum.Options | useLocking (booleano useLocking) |
Estático ResourceSparseApplyKerasMomentum.Options | useNesterov (booleano useNesterov) |
Métodos Heredados
Métodos públicos
public static ResourceSparseApplyKerasMomentum create ( Ámbito alcance, Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <U> í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 |
---|---|
variable | Debe ser de una Variable(). |
acumular | Debe ser de una Variable(). |
yo | 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
público estático ResourceSparseApplyKerasMomentum.Options useLocking (booleano useLocking)
Parámetros
usarBloqueo | Si `True`, la actualización de los tensores var y accum estará protegida por un candado; de lo contrario, el comportamiento no está definido, pero puede exhibir menos contención. |
---|
público estático ResourceSparseApplyKerasMomentum.Options useNesterov (booleano useNesterov)
Parámetros
usarNésterov | Si `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. |
---|