SparseApplyMomentum

SparseApplyMomentum 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:

$$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>
crear ( Alcance alcance, operando <T> var, operando <T> 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 SparseApplyMomentum.
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

Valor constante: "SparseApplyMomentum"

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

pública de salida <T> a cabo ()

Igual que "var".

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.