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:
$$accum = accum * momentum + grad$$$$var -= lr * accum$$
Clases anidadas
clase | SparseApplyMomentum.Options | Los atributos opcionales para SparseApplyMomentum |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico del tensor. |
estática <T se extiende Ttype > SparseApplyMomentum <T> | |
Salida <T> | fuera () Igual que "var". |
estáticas SparseApplyMomentum.Options | useLocking (Boolean useLocking) |
estáticas SparseApplyMomentum.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
pública de salida <T> asOutput ()
Devuelve el identificador simbólico del tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static SparseApplyMomentum <T> Crear ( Alcance alcance, operando <T> var, operando <T> Acum, operando <T> LR, operando <T> graduado, operando <? extiende TNumber > índices, operando <T> ímpetu, Opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación SparseApplyMomentum.
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 SparseApplyMomentum
estáticas pública SparseApplyMomentum.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. |
---|
estáticas pública SparseApplyMomentum.Options useNesterov (booleano useNesterov)
Parámetros
useNesterov | Si es "Verdadero", el tensor pasado para calcular grad será var - lr * momentum * acum, por lo que al final, la var que obtienes es en realidad var - lr * momentum * acum. |
---|