ResourceApplyKerasMomentum

classe finale pubblica ResourceApplyKerasMomentum

Aggiorna '* var' in base allo schema momentum.

Impostare use_nesterov = True se si desidera utilizzare lo slancio di Nesterov.

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

Classi annidate

classe ResourceApplyKerasMomentum.Options Attributi facoltativi per ResourceApplyKerasMomentum

Metodi pubblici

static <T> ResourceApplyKerasMomentum
create ( ambito ambito, operando <?> var, operando <?> accum, operando <T> lr, operando <T> grad, operando <T> momentum, opzioni ... opzioni)
Metodo Factory per creare una classe che racchiude una nuova operazione ResourceApplyKerasMomentum.
statico ResourceApplyKerasMomentum.Options
useLocking (booleano useLocking)
statico ResourceApplyKerasMomentum.Options
useNesterov (booleano useNesterov)

Metodi ereditati

Metodi pubblici

public static ResourceApplyKerasMomentum create ( ambito ambito, operando <?> var, operando <?> accum, operando <T> lr, operando <T> grad, operando <T> momentum, opzioni ... opzioni)

Metodo Factory per creare una classe che racchiude una nuova operazione ResourceApplyKerasMomentum.

Parametri
scopo ambito attuale
var Dovrebbe provenire da una variabile ().
accum Dovrebbe provenire da una variabile ().
lr Fattore di scala. Deve essere uno scalare.
grad Il gradiente.
quantità di moto Quantità di moto. Deve essere uno scalare.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di ResourceApplyKerasMomentum

public static ResourceApplyKerasMomentum.Options useLocking (Boolean useLocking)

Parametri
useLocking Se "True", l'aggiornamento dei tensori var e accum sarà protetto da un lock; in caso contrario, il comportamento è indefinito, ma potrebbe presentare meno contese.

public static ResourceApplyKerasMomentum.Options useNesterov (Boolean useNesterov)

Parametri
useNesterov Se "True", il tensore passato per calcolare grad sarà var + momentum * accum, quindi alla fine la var che ottieni è effettivamente var + momentum * accum.