SparseApplyMomentum

classe final pública SparseApplyMomentum

Atualize as entradas relevantes em '*var' e '*accum' de acordo com o esquema de impulso.

Defina use_nesterov = True se quiser usar o impulso Nesterov.

Ou seja, para as linhas para as quais graduamos, atualizamos var e accum da seguinte forma:

$$accum = accum * momentum + grad$$$$var -= lr * accum$$

Classes aninhadas

aula SparseApplyMomentum.Options Atributos opcionais para SparseApplyMomentum

Constantes

Corda OP_NAME O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico do tensor.
estático <T estende TType > SparseApplyMomentum <T>
create ( Escopo do escopo , Operando <T> var, Operando <T> accum, Operando <T> lr, Operando <T> grad, Operando <? estende TNumber > índices, Operando <T> momentum, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação SparseApplyMomentum.
Saída <T>
fora ()
O mesmo que "var".
SparseApplyMomentum.Options estático
useLocking (booleano useLocking)
SparseApplyMomentum.Options estático
useNesterov (booleano useNesterov)

Métodos herdados

Constantes

String final estática pública OP_NAME

O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow

Valor constante: "SparseApplyMomentum"

Métodos Públicos

Saída pública <T> asOutput ()

Retorna o identificador simbólico do tensor.

As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static SparseApplyMomentum <T> create ( Escopo escopo , Operando <T> var, Operando <T> accum, Operando <T> lr, Operando <T> grad, Operando <? estende TNumber > índices, Operando <T> momentum, Opções ... opções)

Método de fábrica para criar uma classe que envolve uma nova operação SparseApplyMomentum.

Parâmetros
escopo escopo atual
var Deve ser de uma variável().
acumular Deve ser de uma variável().
lr Taxa de Aprendizagem. Deve ser um escalar.
graduado O gradiente.
índices Um vetor de índices na primeira dimensão de var e accum.
impulso Momento. Deve ser um escalar.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de SparseApplyMomentum

Saída pública <T> out ()

O mesmo que "var".

público estático SparseApplyMomentum.Options useLocking (booleano useLocking)

Parâmetros
useLocking Se `True`, a atualização dos tensores var e accum será protegida por um bloqueio; caso contrário, o comportamento será indefinido, mas poderá apresentar menos contenção.

public static SparseApplyMomentum.Options useNesterov (booleano useNesterov)

Parâmetros
usarNesterov Se `True`, o tensor passado para calcular grad será var - lr * momentum * accum, então, no final, o var que você obtém é na verdade var - lr * momentum * accum.