Aggiorna '*var' secondo lo schema del momentum.
Imposta use_nesterov = True se desideri utilizzare lo slancio di Nesterov.
accum = accum * momento + grad var -= lr * accum
Classi nidificate
classe | ApplyMomentum.Options | Attributi facoltativi per ApplyMomentum |
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
Uscita <T> | comeuscita () Restituisce l'handle simbolico del tensore. |
statico <T estende TType > ApplyMomentum <T> | |
Uscita <T> | fuori () Uguale a "var". |
ApplyMomentum.Options statico | useLocking (useLocking booleano) |
ApplyMomentum.Options statico | useNesterov (uso booleanoNesterov) |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
Uscita pubblica <T> asOutput ()
Restituisce l'handle simbolico del tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
public static ApplyMomentum <T> create ( Scope scope, Operando <T> var, Operando <T> accum, Operando <T> lr, Operando <T> grad, Operando <T> momentum, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione ApplyMomentum.
Parametri
scopo | ambito attuale |
---|---|
var | Dovrebbe provenire da una variabile(). |
accu | Dovrebbe provenire da una variabile(). |
lr | Fattore di scala. Deve essere uno scalare. |
grado | Il gradiente. |
quantità di moto | Quantità di moto. Deve essere uno scalare. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di ApplyMomentum
public static ApplyMomentum.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 ApplyMomentum.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. |
---|