ResourceSparseApplyKerasMomentum

classe finale pubblica ResourceSparseApplyKerasMomentum

Aggiorna le voci rilevanti in '*var' e '*accum' secondo lo schema del momentum.

Imposta use_nesterov = True se desideri utilizzare lo slancio di Nesterov.

Questo è per le righe per le quali abbiamo grad, aggiorniamo var e accum come segue:

accum = accum * quantità di moto - lr * grad var += accum

Classi nidificate

classe ResourceSparseApplyKerasMomentum.Options Attributi facoltativi per ResourceSparseApplyKerasMomentum

Metodi pubblici

statico <T, U estende Numero> ResourceSparseApplyKerasMomentum
create ( Scope scope, Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <U> indici, Operando <T> momentum, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione ResourceSparseApplyKerasMomentum.
static ResourceSparseApplyKerasMomentum.Options
useLocking (useLocking booleano)
static ResourceSparseApplyKerasMomentum.Options
useNesterov (uso booleanoNesterov)

Metodi ereditati

Metodi pubblici

public static ResourceSparseApplyKerasMomentum create ( Scope scope, Operando <?> var, Operando <?> accum, Operando <T> lr, Operando <T> grad, Operando <U> indici, Operando <T> momentum, Opzioni... opzioni)

Metodo factory per creare una classe che racchiude una nuova operazione ResourceSparseApplyKerasMomentum.

Parametri
scopo ambito attuale
var Dovrebbe provenire da una variabile().
accu Dovrebbe provenire da una variabile().
lr Tasso di apprendimento. Deve essere uno scalare.
grado Il gradiente.
indici Un vettore di indici nella prima dimensione di var e accum.
quantità di moto Quantità di moto. Deve essere uno scalare.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di ResourceSparseApplyKerasMomentum

public static ResourceSparseApplyKerasMomentum.Options useLocking (useLocking booleano)

Parametri
utilizzareBlocco Se "Vero", l'aggiornamento dei tensori var e accum sarà protetto da un blocco; altrimenti il ​​comportamento non è definito, ma può mostrare meno contesa.

public static ResourceSparseApplyKerasMomentum.Options useNesterov (uso booleanoNesterov)

Parametri
utilizzare Nesterov Se "True", il tensore passato al calcolo grad sarà var + momentum * accum, quindi alla fine, il var che ottieni è in realtà var + momentum * accum.