ResourceSparseApplyMomentum

classe finale pubblica ResourceSparseApplyMomentum

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 * momento + grad var -= lr * accum

Classi nidificate

classe ResourceSparseApplyMomentum.Options Attributi facoltativi per ResourceSparseApplyMomentum

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

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

Metodi ereditati

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "ResourceSparseApplyMomentum"

Metodi pubblici

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

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

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 ResourceSparseApplyMomentum

public static ResourceSparseApplyMomentum.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 ResourceSparseApplyMomentum.Options useNesterov (booleano useNesterov)

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