ResourceSparseApplyKerasMomentum

ResourceSparseApplyKerasMomentum public final class

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:

acum = acumular * impulso - lr * grad var + = acumular

Clases anidadas

clase ResourceSparseApplyKerasMomentum.Options Los atributos opcionales para ResourceSparseApplyKerasMomentum

Métodos públicos

estática <T, U se extiende Número> ResourceSparseApplyKerasMomentum
crear ( Alcance alcance, operando <?> var, operando <?> acum, operando <T> lr, operando <T> graduado, operando <U> índices, operando <T> impulso, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación ResourceSparseApplyKerasMomentum.
estáticas ResourceSparseApplyKerasMomentum.Options
useLocking (Boolean useLocking)
estáticas ResourceSparseApplyKerasMomentum.Options
useNesterov (Boolean useNesterov)

Métodos heredados

Métodos públicos

public static ResourceSparseApplyKerasMomentum crear ( Alcance alcance, operando <?> var, operando <?> acum, operando <T> lr, operando <T> graduado, operando <U> í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 mostrar 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, la var que obtienes es en realidad var + momentum * acum.