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 | |
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. |
---|