Perbarui '*var' sesuai dengan skema momentum.
Tetapkan use_nesterov = True jika Anda ingin menggunakan momentum Nesterov.
accum = accum * momentum + lulusan var -= lr * accum
Kelas Bersarang
| kelas | TerapkanMomentum.Opsi | Atribut opsional untuk ApplyMomentum | |
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
| Keluaran <T> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
| statis <T memperluas TType > ApplyMomentum <T> | |
| Keluaran <T> | keluar () Sama seperti "var". |
| ApplyMomentum.Options statis | useLocking (penguncian penggunaan Boolean) |
| ApplyMomentum.Options statis | useNesterov (Boolean useNesterov) |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static ApplyMomentum <T> buat ( Lingkup lingkup, Operand <T> var, Operand <T> accum, Operand <T> lr, Operand <T> grad, Operand <T> momentum, Opsi... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi ApplyMomentum baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|---|
| var | Harus dari Variabel(). |
| akumulasi | Harus dari Variabel(). |
| lr | Faktor skala. Pasti skalar. |
| lulusan | Gradien. |
| momentum | momentum. Pasti skalar. |
| pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari ApplyMomentum
ApplyMomentum.Options useLocking statis publik (Boolean useLocking)
Parameter
| gunakan Penguncian | Jika `True`, pembaruan tensor var dan accum akan dilindungi oleh kunci; jika tidak, perilaku tersebut tidak terdefinisikan, namun mungkin menunjukkan lebih sedikit pertentangan. |
|---|
ApplyMomentum.Options statis publik useNesterov (Boolean useNesterov)
Parameter
| gunakanNesterov | Jika `True`, tensor yang diteruskan ke komputasi grad adalah var - lr * momentum * accum, jadi pada akhirnya, var yang Anda dapatkan sebenarnya adalah var - lr * momentum * accum. |
|---|